Spaces:
Running
on
Zero
Running
on
Zero
Commit
Β·
620a643
1
Parent(s):
ba6a88b
yes
Browse files
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
|
10 |
-
from
|
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 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")
|