Spaces:

J-LAB
/

FluxiIA_LLama-Florence

Running on Zero

J-LAB commited on Nov 13, 2024

Commit

96c662b

verified ·

1 Parent(s): 196a100

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -9,7 +9,7 @@ import numpy as np
 import subprocess
 import os
-subprocess.run('pip install flash-attn --no-build-isolation', env={'FLASH_ATTENTION_SKIP_CUDA_BUILD': "TRUE"}, shell=True)
 device = "cuda" if torch.cuda.is_available() else "cpu"
 model_id = 'J-LAB/Florence-vl3'

 import subprocess
 import os
+#subprocess.run('pip install flash-attn --no-build-isolation', env={'FLASH_ATTENTION_SKIP_CUDA_BUILD': "TRUE"}, shell=True)
 device = "cuda" if torch.cuda.is_available() else "cpu"
 model_id = 'J-LAB/Florence-vl3'