Spaces:

IDEA-CCNL
/

Ziya-BLIP2-14B-Visual-v1-Demo

Runtime error

wuxiaojun commited on Jul 18, 2023

Commit

1b28cbf

1 Parent(s): d9c12fa

指定BNB_CUDA_VERSION in bitsandbytes

Files changed (1) hide show

app.py CHANGED Viewed

@@ -2,6 +2,6 @@ import os
 os.system(
     f"git clone https://github.com/TimDettmers/bitsandbytes.git /home/user/app/bitsandbytes")
-os.system(f"export BNB_CUDA_VERSION=113")
 os.system(f"cd /home/user/app/bitsandbytes && CUDA_VERSION=113 make cuda11x && python setup.py install")
 os.system(f"python /home/user/app/launch.py")

 os.system(
     f"git clone https://github.com/TimDettmers/bitsandbytes.git /home/user/app/bitsandbytes")
+os.system(f"export BNB_CUDA_VERSION=117")
 os.system(f"cd /home/user/app/bitsandbytes && CUDA_VERSION=113 make cuda11x && python setup.py install")
 os.system(f"python /home/user/app/launch.py")