Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -4,6 +4,7 @@ subprocess.run(
|
|
4 |
env={'FLASH_ATTENTION_SKIP_CUDA_BUILD': "TRUE"},
|
5 |
shell=True
|
6 |
)
|
|
|
7 |
from threading import Thread
|
8 |
import torch
|
9 |
from PIL import Image
|
|
|
4 |
env={'FLASH_ATTENTION_SKIP_CUDA_BUILD': "TRUE"},
|
5 |
shell=True
|
6 |
)
|
7 |
+
subprocess.run(["pip", "install", "-i", "https://pypi.org/simple/", "bitsandbytes"], check=True)
|
8 |
from threading import Thread
|
9 |
import torch
|
10 |
from PIL import Image
|