Spaces:
Runtime error
Runtime error
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
|