VictorSanh commited on
Commit
aa7cdbb
·
1 Parent(s): 4273b28

temporary hack

Browse files
Files changed (1) hide show
  1. app_dialogue.py +2 -0
app_dialogue.py CHANGED
@@ -5,6 +5,8 @@ import subprocess
5
  subprocess.run('pip install flash-attn --no-build-isolation', env={'FLASH_ATTENTION_SKIP_CUDA_BUILD': "TRUE"}, shell=True)
6
  # Install private transformers fork which is the only place where idefics2 has been integrated at the time being
7
  subprocess.run(f"pip install git+https://VictorSanh:{os.environ['TRANSFORMERS_NEW_MODEL_ADDITION_TOKEN']}@github.com/huggingface/new-model-addition.git@b122cb2d878159a2928a9b423b157126f5f89ed4", shell=True)
 
 
8
 
9
 
10
  import copy
 
5
  subprocess.run('pip install flash-attn --no-build-isolation', env={'FLASH_ATTENTION_SKIP_CUDA_BUILD': "TRUE"}, shell=True)
6
  # Install private transformers fork which is the only place where idefics2 has been integrated at the time being
7
  subprocess.run(f"pip install git+https://VictorSanh:{os.environ['TRANSFORMERS_NEW_MODEL_ADDITION_TOKEN']}@github.com/huggingface/new-model-addition.git@b122cb2d878159a2928a9b423b157126f5f89ed4", shell=True)
8
+ # Hack once we figure zero error
9
+ subprocess.run(f"pip install spaces==0.25.0", shell=True)
10
 
11
 
12
  import copy