Spaces:
Running
Running
Update app.py with improved Gradio interface and add deepspeed to requirements.txt
Browse files- app.py +0 -1
- requirements.txt +3 -1
app.py
CHANGED
|
@@ -1,5 +1,4 @@
|
|
| 1 |
import gradio as gr
|
| 2 |
-
import torch
|
| 3 |
from transformers import AutoModel, AutoTokenizer
|
| 4 |
|
| 5 |
device = "cpu"
|
|
|
|
| 1 |
import gradio as gr
|
|
|
|
| 2 |
from transformers import AutoModel, AutoTokenizer
|
| 3 |
|
| 4 |
device = "cpu"
|
requirements.txt
CHANGED
|
@@ -2,4 +2,6 @@ Pillow==10.1.0
|
|
| 2 |
torch==2.1.2
|
| 3 |
torchvision==0.16.2
|
| 4 |
transformers==4.40.0
|
| 5 |
-
|
|
|
|
|
|
|
|
|
| 2 |
torch==2.1.2
|
| 3 |
torchvision==0.16.2
|
| 4 |
transformers==4.40.0
|
| 5 |
+
deepspeed==0.12.6
|
| 6 |
+
sentencepiece==0.1.99
|
| 7 |
+
gradio
|