khazarai
/

Math-VL-8B

Image-Text-to-Text

text-generation-inference

Model card Files Files and versions

Rustamshry commited on 9 days ago

Commit

e710cbc

·

verified ·

1 Parent(s): ca62d0e

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -35,6 +35,6 @@ inputs = processor.apply_chat_template(
 	return_tensors="pt",
 ).to(model.device)
-outputs = model.generate(**inputs, max_new_tokens=40)
 print(processor.decode(outputs[0][inputs["input_ids"].shape[-1]:]))
 ```

 	return_tensors="pt",
 ).to(model.device)
+outputs = model.generate(**inputs, max_new_tokens=1024)
 print(processor.decode(outputs[0][inputs["input_ids"].shape[-1]:]))
 ```