recoilme commited on
Commit
fdbe333
·
verified ·
1 Parent(s): 6f1763d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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, "vae"):
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)