Update README.md
Browse files
README.md
CHANGED
|
@@ -47,7 +47,7 @@ model_id = "OpenVINO/starcoder2-15b-int4-ov"
|
|
| 47 |
tokenizer = AutoTokenizer.from_pretrained(model_id)
|
| 48 |
model = OVModelForCausalLM.from_pretrained(model_id)
|
| 49 |
|
| 50 |
-
inputs = tokenizer("
|
| 51 |
|
| 52 |
outputs = model.generate(**inputs, max_length=200)
|
| 53 |
text = tokenizer.batch_decode(outputs)[0]
|
|
|
|
| 47 |
tokenizer = AutoTokenizer.from_pretrained(model_id)
|
| 48 |
model = OVModelForCausalLM.from_pretrained(model_id)
|
| 49 |
|
| 50 |
+
inputs = tokenizer("def print_hello_world():", return_tensors="pt")
|
| 51 |
|
| 52 |
outputs = model.generate(**inputs, max_length=200)
|
| 53 |
text = tokenizer.batch_decode(outputs)[0]
|