Spaces:

zamal
/

Molmo-4bit

Running on Zero

zamal commited on Oct 2, 2024

Commit

b62fd98

verified ·

1 Parent(s): c349d1e

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -6,8 +6,8 @@ import os
 import subprocess
 subprocess.run('pip install flash-attn --no-build-isolation', env={'FLASH_ATTENTION_SKIP_CUDA_BUILD': "TRUE"}, shell=True)
-subprocess.run('pip install -U bitsandbytes')
 # Define the repository for the quantized model

 import subprocess
 subprocess.run('pip install flash-attn --no-build-isolation', env={'FLASH_ATTENTION_SKIP_CUDA_BUILD': "TRUE"}, shell=True)
+os.system('pip install -U bitsandbytes')
 # Define the repository for the quantized model