HuggingFaceM4
/

idefics2-8b-base

Image-Text-to-Text

text-generation-inference

Inference Endpoints

Model card Files Files and versions Community

VictorSanh commited on Apr 13

Commit

65215b0

•

1 Parent(s): dff8ac8

`eos_token_id`

Files changed (1) hide show

README.md +2 -2

README.md CHANGED Viewed

@@ -158,7 +158,7 @@ inputs = {k: v.to(DEVICE) for k, v in inputs.items()}
 # Generate
-generated_ids = model.generate(**inputs, bad_words_ids=BAD_WORDS_IDS, max_new_tokens=500)
 generated_texts = processor.batch_decode(generated_ids, skip_special_tokens=True)
 print(generated_texts)
@@ -209,7 +209,7 @@ inputs = {k: v.to(DEVICE) for k, v in inputs.items()}
 # Generate
-generated_ids = model.generate(**inputs, bad_words_ids=BAD_WORDS_IDS, max_new_tokens=500)
 generated_texts = processor.batch_decode(generated_ids, skip_special_tokens=True)
 print(generated_texts)

 # Generate
+generated_ids = model.generate(**inputs, bad_words_ids=BAD_WORDS_IDS, eos_token_id=EOS_WORDS_IDS, max_new_tokens=500)
 generated_texts = processor.batch_decode(generated_ids, skip_special_tokens=True)
 print(generated_texts)
 # Generate
+generated_ids = model.generate(**inputs, bad_words_ids=BAD_WORDS_IDS, eos_token_id=EOS_WORDS_IDS, max_new_tokens=500)
 generated_texts = processor.batch_decode(generated_ids, skip_special_tokens=True)
 print(generated_texts)