Spaces:
Running
Running
Update app.py
Browse files
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-
|
29 |
-
pipe.text_encoder = PeftModel.from_pretrained(pipe.text_encoder, "AnastasiaSh/sticker-cat-
|
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}")
|