Aria

Running on Zero

andito HF staff commited on Nov 18

Commit

aab352a

•

1 Parent(s): d24c16c

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -16,7 +16,7 @@ model = Idefics3ForConditionalGeneration.from_pretrained("HuggingFaceTB/SmolVLM_
         #_attn_implementation="flash_attention_2"
                                                         ).to("cuda")
-BAD_WORDS_IDS = processor.tokenizer(["<image>", "<fake_token_around_image>"], add_special_tokens=False).input_ids
 EOS_WORDS_IDS = [processor.tokenizer.eos_token_id]
 @spaces.GPU

         #_attn_implementation="flash_attention_2"
                                                         ).to("cuda")
+BAD_WORDS_IDS = processor.tokenizer(["<image>", "<fake_token_around_image>", "<row_", "apiro", "\u2500lrow_", "row_1"], add_special_tokens=False).input_ids
 EOS_WORDS_IDS = [processor.tokenizer.eos_token_id]
 @spaces.GPU