idefics2_playground

Running on Zero

Leyo commited on Apr 18

Commit

9e85d8f

•

1 Parent(s): 13fd3a2

fix flash

Files changed (1) hide show

app_dialogue.py CHANGED Viewed

@@ -2,7 +2,7 @@ import os
 import subprocess
 # Install flash attention
-# subprocess.run('pip install flash-attn --no-build-isolation', env={'FLASH_ATTENTION_SKIP_CUDA_BUILD': "TRUE"}, shell=True)
 import copy

 import subprocess
 # Install flash attention
+subprocess.run('pip install flash-attn --no-build-isolation', env={'FLASH_ATTENTION_SKIP_CUDA_BUILD': "TRUE"}, shell=True)
 import copy