Spaces:
Build error
Build error
Update model.py
Browse files
model.py
CHANGED
@@ -7,6 +7,8 @@ Original file is located at
|
|
7 |
https://colab.research.google.com/drive/1_vTVH3hBX8wVXIgrW1T2Q4N1DSkWoXV8
|
8 |
"""
|
9 |
|
|
|
|
|
10 |
|
11 |
|
12 |
import gradio as gr
|
@@ -14,7 +16,9 @@ import torch
|
|
14 |
from transformers import TextStreamer
|
15 |
from unsloth import FastLanguageModel
|
16 |
from google.colab import drive
|
17 |
-
|
|
|
|
|
18 |
|
19 |
# Ensure necessary packages are installed
|
20 |
|
|
|
7 |
https://colab.research.google.com/drive/1_vTVH3hBX8wVXIgrW1T2Q4N1DSkWoXV8
|
8 |
"""
|
9 |
|
10 |
+
import os
|
11 |
+
os.system('pip install "unsloth[colab-new] @ git+https://github.com/unslothai/unsloth.git" --no-deps xformers trl peft accelerate bitsandbytes')
|
12 |
|
13 |
|
14 |
import gradio as gr
|
|
|
16 |
from transformers import TextStreamer
|
17 |
from unsloth import FastLanguageModel
|
18 |
from google.colab import drive
|
19 |
+
|
20 |
+
|
21 |
+
|
22 |
|
23 |
# Ensure necessary packages are installed
|
24 |
|