trashchenkov commited on
Commit
936c0a6
·
verified ·
1 Parent(s): ba95d06

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -25,8 +25,8 @@ pipe = DiffusionPipeline.from_pretrained(model_repo_id, torch_dtype=torch_dtype)
25
 
26
  # Попробуем подгрузить LoRA-модификации (unet + text_encoder)
27
  try:
28
- pipe.unet = PeftModel.from_pretrained(pipe.unet, "AnastasiaSh/sticker-cat-lora3/unet")
29
- pipe.text_encoder = PeftModel.from_pretrained(pipe.text_encoder, "AnastasiaSh/sticker-cat-lora3/text_encoder")
30
  except Exception as e:
31
  # Если не удалось, можно вывести предупреждение или поднять ошибку
32
  print(f"Не удалось подгрузить LoRA по умолчанию: {e}")
 
25
 
26
  # Попробуем подгрузить LoRA-модификации (unet + text_encoder)
27
  try:
28
+ pipe.unet = PeftModel.from_pretrained(pipe.unet, "AnastasiaSh/sticker-cat-lora4/unet")
29
+ pipe.text_encoder = PeftModel.from_pretrained(pipe.text_encoder, "AnastasiaSh/sticker-cat-lora4/text_encoder")
30
  except Exception as e:
31
  # Если не удалось, можно вывести предупреждение или поднять ошибку
32
  print(f"Не удалось подгрузить LoRA по умолчанию: {e}")