Update app.py

app.py

@@ -10,7 +10,7 @@ import numpy as np
 import os
 
 HF_TOKEN = os.environ['HF_TOKEN']
-subprocess.run('pip install flash-attn --no-build-isolation', env={'FLASH_ATTENTION_SKIP_CUDA_BUILD': "TRUE"}, shell=True)
+# subprocess.run('pip install flash-attn --no-build-isolation', env={'FLASH_ATTENTION_SKIP_CUDA_BUILD': "TRUE"}, shell=True)
 
 # models = {
 #     "Qwen/Qwen2-VL-2B-Instruct": AutoModelForCausalLM.from_pretrained("Qwen/Qwen2-VL-2B-Instruct", trust_remote_code=True, torch_dtype="auto", _attn_implementation="flash_attention_2").cuda().eval()
@@ -38,7 +38,7 @@ models = {
         trust_remote_code=True, 
         token=HF_TOKEN,
         torch_dtype=torch.bfloat16,
-        attn_implementation="flash_attention_2"
+        # attn_implementation="flash_attention_2"
     ).cuda().eval()
 
 }