Hugging Face's logo

Spaces:
IDEA-CCNL
/
Ziya-BLIP2-14B-Visual-v1-Demo
Runtime error

App Files Community
3
wuxiaojun commited on
Commit
7a815a8
•
1 Parent(s): 1b28cbf

指定BNB_CUDA_VERSION in bitsandbytes

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -2,6 +2,6 @@ import os
2
 
3
  os.system(
4
  f"git clone https://github.com/TimDettmers/bitsandbytes.git /home/user/app/bitsandbytes")
5
- os.system(f"export BNB_CUDA_VERSION=117")
6
  os.system(f"cd /home/user/app/bitsandbytes && CUDA_VERSION=113 make cuda11x && python setup.py install")
7
  os.system(f"python /home/user/app/launch.py")
 
2
 
3
  os.system(
4
  f"git clone https://github.com/TimDettmers/bitsandbytes.git /home/user/app/bitsandbytes")
5
+ os.system(f"export LD_LIBRARY_PATH=/usr/local/cuda:$LD_LIBRARY_PATH && export BNB_CUDA_VERSION=113")
6
  os.system(f"cd /home/user/app/bitsandbytes && CUDA_VERSION=113 make cuda11x && python setup.py install")
7
  os.system(f"python /home/user/app/launch.py")