Spaces:

anand004
/

Multimodal-PDF-RAG

Runtime error

anand004 commited on Jun 24, 2024

Commit

1aaa4a1

verified ·

1 Parent(s): c1f0801

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -53,7 +53,6 @@ if torch.cuda.is_available():
         torch_dtype=torch.float16,
         low_cpu_mem_usage=True,
         load_in_4bit=True,
-        use_flash_attention_2=True
     )

         torch_dtype=torch.float16,
         low_cpu_mem_usage=True,
         load_in_4bit=True,
     )