OpenVINO
/

starcoder2-7b-int4-ov

Text Generation

Inference Endpoints

Model card Files Files and versions Community

katuni4ka commited on Jun 20

Commit

714ce36

•

1 Parent(s): c0114c0

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -47,7 +47,7 @@ model_id = "OpenVINO/starcoder2-7b-int4-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]