Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
|
@@ -14,7 +14,7 @@ DTYPE = torch.float16
|
|
| 14 |
|
| 15 |
# ---- Загрузка модели ----
|
| 16 |
def load_vae(model_id=MODEL_ID, device=DEVICE):
|
| 17 |
-
for attempt in (None, "
|
| 18 |
try:
|
| 19 |
if attempt is None:
|
| 20 |
vae = AsymmetricAutoencoderKL.from_pretrained(model_id, torch_dtype=DTYPE)
|
|
|
|
| 14 |
|
| 15 |
# ---- Загрузка модели ----
|
| 16 |
def load_vae(model_id=MODEL_ID, device=DEVICE):
|
| 17 |
+
for attempt in (None, "vae4"):
|
| 18 |
try:
|
| 19 |
if attempt is None:
|
| 20 |
vae = AsymmetricAutoencoderKL.from_pretrained(model_id, torch_dtype=DTYPE)
|