Hugging Face's logo

Spaces:
zamal
/
Molmo-4bit
Running on Zero

App Files Community
zamal commited on
Commit
f82d3e1
·
verified ·
1 Parent(s): 7c42a56

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -6,7 +6,7 @@ import os
6
  import subprocess
7
  subprocess.run('pip install flash-attn --no-build-isolation', env={'FLASH_ATTENTION_SKIP_CUDA_BUILD': "TRUE"}, shell=True)
8
 
9
- os.system('pip install -U bitsandbytes')
10
 
11
 
12
 
 
6
  import subprocess
7
  subprocess.run('pip install flash-attn --no-build-isolation', env={'FLASH_ATTENTION_SKIP_CUDA_BUILD': "TRUE"}, shell=True)
8
 
9
+ os.system('pip install -U bitsandbytes-cuda117')
10
 
11
 
12