Spaces:

arad1367
/

Multimodal_RAG_Pejman

App Files Files Community

arad1367 commited on 9 days ago

Commit

d620d8e

•

1 Parent(s): 1c38311

Update app.py

Files changed (1) hide show

app.py +1 -1

app.py CHANGED Viewed

@@ -21,7 +21,7 @@ install_poppler()
 subprocess.run('pip install flash-attn --no-build-isolation', env={'FLASH_ATTENTION_SKIP_CUDA_BUILD': "TRUE"}, shell=True)
-RAG = RAGMultiModalModel.from_pretrained("vidore/colpali")
 model = Qwen2VLForConditionalGeneration.from_pretrained("Qwen/Qwen2-VL-2B-Instruct",
                                                         trust_remote_code=True, torch_dtype=torch.bfloat16).cuda().eval()
 processor = AutoProcessor.from_pretrained("Qwen/Qwen2-VL-2B-Instruct", trust_remote_code=True)

 subprocess.run('pip install flash-attn --no-build-isolation', env={'FLASH_ATTENTION_SKIP_CUDA_BUILD': "TRUE"}, shell=True)
+RAG = RAGMultiModalModel.from_pretrained("vidore/colpali-v1.2")
 model = Qwen2VLForConditionalGeneration.from_pretrained("Qwen/Qwen2-VL-2B-Instruct",
                                                         trust_remote_code=True, torch_dtype=torch.bfloat16).cuda().eval()
 processor = AutoProcessor.from_pretrained("Qwen/Qwen2-VL-2B-Instruct", trust_remote_code=True)