OpenVINO
/

starcoder2-7b-fp16-ov

Text Generation

Inference Endpoints

Model card Files Files and versions Community

katuni4ka commited on Jun 20

Commit

58b84ea

•

1 Parent(s): e0491b8

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -36,7 +36,7 @@ model_id = "OpenVINO/starcoder2-7b-fp16-ov"
 tokenizer = AutoTokenizer.from_pretrained(model_id)
 model = OVModelForCausalLM.from_pretrained(model_id)
-inputs = tokenizer("What is OpenVINO?", return_tensors="pt")
 outputs = model.generate(**inputs, max_length=200)
 text = tokenizer.batch_decode(outputs)[0]

 tokenizer = AutoTokenizer.from_pretrained(model_id)
 model = OVModelForCausalLM.from_pretrained(model_id)
+inputs = tokenizer("def print_hello_world():", return_tensors="pt")
 outputs = model.generate(**inputs, max_length=200)
 text = tokenizer.batch_decode(outputs)[0]