idefics_and_chatty

Runtime error

Leyo commited on Apr 18, 2024

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