Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -496,11 +496,12 @@ def infer(
|
|
496 |
|
497 |
if use_LCM_adapter == True:
|
498 |
|
499 |
-
print('use_LCM_adapter = '
|
500 |
|
501 |
if LCM_adapter == "lcm-lora-sdv1-5":
|
502 |
adapter_id = "latent-consistency/lcm-lora-sdv1-5"
|
503 |
|
|
|
504 |
generator = torch.Generator(device).manual_seed(seed)
|
505 |
|
506 |
pipe_LCM = get_lora_sd_pipeline(lora_dir='lora_man_animestyle', base_model_name_or_path=model_default, dtype=torch_dtype).to(device)
|
@@ -530,7 +531,7 @@ def infer(
|
|
530 |
|
531 |
if use_DDIMScheduler == True:
|
532 |
|
533 |
-
print('use_DDIMScheduler = '
|
534 |
|
535 |
generator = torch.Generator(device).manual_seed(seed)
|
536 |
|
@@ -555,6 +556,8 @@ def infer(
|
|
555 |
else:
|
556 |
# Генерация изображений с LORA без ControlNet и IP_Adapter ---------------------------------------------------------------------------------------------
|
557 |
|
|
|
|
|
558 |
# Инициализация ControlNet
|
559 |
controlnet_model_path = "lllyasviel/sd-controlnet-openpose"
|
560 |
controlnet = ControlNetModel.from_pretrained(controlnet_model_path, torch_dtype=torch_dtype)
|
|
|
496 |
|
497 |
if use_LCM_adapter == True:
|
498 |
|
499 |
+
print('use_LCM_adapter = True')
|
500 |
|
501 |
if LCM_adapter == "lcm-lora-sdv1-5":
|
502 |
adapter_id = "latent-consistency/lcm-lora-sdv1-5"
|
503 |
|
504 |
+
guidance_scale = 0
|
505 |
generator = torch.Generator(device).manual_seed(seed)
|
506 |
|
507 |
pipe_LCM = get_lora_sd_pipeline(lora_dir='lora_man_animestyle', base_model_name_or_path=model_default, dtype=torch_dtype).to(device)
|
|
|
531 |
|
532 |
if use_DDIMScheduler == True:
|
533 |
|
534 |
+
print('use_DDIMScheduler = True')
|
535 |
|
536 |
generator = torch.Generator(device).manual_seed(seed)
|
537 |
|
|
|
556 |
else:
|
557 |
# Генерация изображений с LORA без ControlNet и IP_Adapter ---------------------------------------------------------------------------------------------
|
558 |
|
559 |
+
print('Генерация изображений с LORA без ControlNet и IP_Adapter')
|
560 |
+
|
561 |
# Инициализация ControlNet
|
562 |
controlnet_model_path = "lllyasviel/sd-controlnet-openpose"
|
563 |
controlnet = ControlNetModel.from_pretrained(controlnet_model_path, torch_dtype=torch_dtype)
|