rag_ColPali_Qwen2VL_7B

Sleeping

AdrienB134 commited on Aug 30

Commit

c785d73

•

1 Parent(s): b728f0b

ruyj

Files changed (1) hide show

app.py CHANGED Viewed

@@ -74,7 +74,7 @@ def model_inference(
     prompt = id_processor.apply_chat_template(resulting_messages, add_generation_prompt=True)
-    inputs = id_processor(text=prompt, images=[images], return_tensors="pt")
     inputs = {k: v.to("cuda") for k, v in inputs.items()}
     generation_args = {

     prompt = id_processor.apply_chat_template(resulting_messages, add_generation_prompt=True)
+    inputs = id_processor(text=prompt, images=images, return_tensors="pt")
     inputs = {k: v.to("cuda") for k, v in inputs.items()}
     generation_args = {