Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -3,14 +3,14 @@ os.system("pip3 install torch==2.2.1 torchvision torchaudio xformers --index-u
|
|
3 |
import torch
|
4 |
major_version, minor_version = torch.cuda.get_device_capability()
|
5 |
# Must install separately since Colab has torch 2.2.1, which breaks packages
|
6 |
-
|
7 |
if major_version >= 8:
|
8 |
# Use this for new GPUs like Ampere, Hopper GPUs (RTX 30xx, RTX 40xx, A100, H100, L40)
|
9 |
-
|
10 |
-
accelerate bitsandbytes
|
11 |
else:
|
12 |
# Use this for older GPUs (V100, Tesla T4, RTX 20xx)
|
13 |
-
|
14 |
pass
|
15 |
|
16 |
|
|
|
3 |
import torch
|
4 |
major_version, minor_version = torch.cuda.get_device_capability()
|
5 |
# Must install separately since Colab has torch 2.2.1, which breaks packages
|
6 |
+
os.system("pip install unsloth[colab-new] @ git+https://github.com/unslothai/unsloth.git")
|
7 |
if major_version >= 8:
|
8 |
# Use this for new GPUs like Ampere, Hopper GPUs (RTX 30xx, RTX 40xx, A100, H100, L40)
|
9 |
+
os.system("pip install --no-deps packaging ninja einops flash-attn xformers trl peft \
|
10 |
+
accelerate bitsandbytes")
|
11 |
else:
|
12 |
# Use this for older GPUs (V100, Tesla T4, RTX 20xx)
|
13 |
+
os.system("pip install --no-deps trl peft accelerate bitsandbytes")
|
14 |
pass
|
15 |
|
16 |
|