trashchenkov commited on
Commit
678010b
·
verified ·
1 Parent(s): e7cd7b2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
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-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}")
@@ -54,8 +54,8 @@ def infer(
54
 
55
  # Повторно подгружаем LoRA для нового пайплайна
56
  try:
57
- new_pipe.unet = PeftModel.from_pretrained(new_pipe.unet, "AnastasiaSh/sticker-cat-lora3/unet")
58
- new_pipe.text_encoder = PeftModel.from_pretrained(new_pipe.text_encoder, "AnastasiaSh/sticker-cat-lora3/text_encoder")
59
  except Exception as e:
60
  raise gr.Error(f"Не удалось подгрузить LoRA: {e}")
61
 
 
25
 
26
  # Попробуем подгрузить LoRA-модификации (unet + text_encoder)
27
  try:
28
+ pipe.unet = PeftModel.from_pretrained(pipe.unet, "./unet")
29
+ pipe.text_encoder = PeftModel.from_pretrained(pipe.text_encoder, "./text_encoder")
30
  except Exception as e:
31
  # Если не удалось, можно вывести предупреждение или поднять ошибку
32
  print(f"Не удалось подгрузить LoRA по умолчанию: {e}")
 
54
 
55
  # Повторно подгружаем LoRA для нового пайплайна
56
  try:
57
+ new_pipe.unet = PeftModel.from_pretrained(new_pipe.unet, "./unet")
58
+ new_pipe.text_encoder = PeftModel.from_pretrained(new_pipe.text_encoder, "./text_encoder")
59
  except Exception as e:
60
  raise gr.Error(f"Не удалось подгрузить LoRA: {e}")
61