Lifeinhockey commited on
Commit
b47703f
·
verified ·
1 Parent(s): e33974e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -2
app.py CHANGED
@@ -496,11 +496,12 @@ def infer(
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
  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 = ', True)
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)