Spaces:
Running
on
Zero
Running
on
Zero
Commit
Β·
d3593ef
1
Parent(s):
35e5214
yes
Browse files- app.py +2 -2
- requirements.txt +2 -1
app.py
CHANGED
@@ -6,8 +6,8 @@ from transformers import T5Tokenizer, T5EncoderModel
|
|
6 |
from diffusers import DiffusionPipeline
|
7 |
from safetensors.torch import load_file
|
8 |
from huggingface_hub import hf_hub_download
|
9 |
-
from two_stream_shunt_adapter import TwoStreamShuntAdapter
|
10 |
-
from configs import T5_SHUNT_REPOS
|
11 |
|
12 |
# βββ Device & Model Setup βββββββββββββββββββββββββββββββββββββ
|
13 |
device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
|
|
|
6 |
from diffusers import DiffusionPipeline
|
7 |
from safetensors.torch import load_file
|
8 |
from huggingface_hub import hf_hub_download
|
9 |
+
from .app.two_stream_shunt_adapter import TwoStreamShuntAdapter
|
10 |
+
from .app.configs import T5_SHUNT_REPOS
|
11 |
|
12 |
# βββ Device & Model Setup βββββββββββββββββββββββββββββββββββββ
|
13 |
device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
|
requirements.txt
CHANGED
@@ -4,4 +4,5 @@ invisible_watermark
|
|
4 |
torch
|
5 |
transformers
|
6 |
xformers
|
7 |
-
matplotlib
|
|
|
|
4 |
torch
|
5 |
transformers
|
6 |
xformers
|
7 |
+
matplotlib
|
8 |
+
sentencepiece
|