PKU-Alignment
/

Beaver-Vision-11B

Image-Text-to-Text

text-generation-inference

Model card Files Files and versions Community

Gaie commited on Nov 8, 2024

Commit

f805d84

·

verified ·

1 Parent(s): 41edc44

Update README.md

Files changed (1) hide show

README.md +6 -1

README.md CHANGED Viewed

@@ -203,7 +203,12 @@ inputs = processor(
 ).to(model.device)
 output = model.generate(**inputs, max_new_tokens=30)
-print(processor.decode(output[0]))
 ```
 ## Citation

 ).to(model.device)
 output = model.generate(**inputs, max_new_tokens=30)
+print(processor.decode(output[0], skip_special_tokens=True))
+# The output:
+# In the garden's embrace,
+# Bunny in a blue coat,
+# Spring's gentle whisper.
 ```
 ## Citation