rag_ColPali_Qwen2VL

Running on Zero

AdrienB134 commited on Aug 30

Commit

ba08945

•

1 Parent(s): 207c30a

jdytnty

Files changed (1) hide show

app.py CHANGED Viewed

@@ -30,8 +30,8 @@ id_model = Idefics3ForConditionalGeneration.from_pretrained("HuggingFaceM4/Idefi
         #_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
 def model_inference(

         #_attn_implementation="flash_attention_2"
                                                         ).to("cuda")
+BAD_WORDS_IDS = id_processor.tokenizer(["<image>", "<fake_token_around_image>"], add_special_tokens=False).input_ids
+EOS_WORDS_IDS = [id_processor.tokenizer.eos_token_id]
 @spaces.GPU
 def model_inference(