| import torch |
| from transformers import AutoModelForCausalLM, AutoTokenizer |
| tokenizer = AutoTokenizer.from_pretrained("stabilityai/stable-code-3b") |
| model = AutoModelForCausalLM.from_pretrained( |
| "stabilityai/stable-code-3b", |
| torch_dtype="auto", |
| ) |
|
|
|
|
| codeSnip = """ |
| Is the following CSS code valid? |
| ``` |
| div { |
| color: red; |
| width: 100px; |
| } |
| ``` |
| """ |
|
|
| inputs = tokenizer(codeSnip, return_tensors="pt") |
|
|
| tokens = model.generate( |
| **inputs, |
| max_new_tokens=48, |
| temperature=0.2, |
| do_sample=True, |
| ) |
|
|
| res = tokenizer.decode(tokens[0], skip_special_tokens=True) |
| print(res) |
|
|
