Update app.py

app.py

@@ -116,21 +116,23 @@ def infer(
     seed=4, 
     guidance_scale=7.5, 
     lora_scale=0.7,
-    use_control_net=False,  # Параметр для включения ControlNet
-    control_mode=None,      # Параметр для выбора режима ControlNet
-    strength_cn=0.5,        # Коэфф. зашумления ControlNet
-    control_strength=0.5,   # Сила влияния ControlNet
-    cn_source_image=None,   # Исходное изображение ControlNet
-    control_image=None,     # Контрольное изображение ControlNet
-    use_ip_adapter=False,   # Параметр для включения IP_adapter
-    ip_adapter_mode=None,   # Параметр для выбора режима IP_adapter
-    strength_ip=0.5,        # Коэфф. зашумления IP_adapter
-    ip_adapter_strength=0.5,# Сила влияния IP_adapter
+    use_control_net=False,   # Параметр для включения ControlNet
+    control_mode=None,       # Параметр для выбора режима ControlNet
+    strength_cn=0.5,         # Коэфф. зашумления ControlNet
+    control_strength=0.5,    # Сила влияния ControlNet
+    cn_source_image=None,    # Исходное изображение ControlNet
+    control_image=None,      # Контрольное изображение ControlNet
+    use_ip_adapter=False,    # Параметр для включения IP_adapter
+    ip_adapter_mode=None,    # Параметр для выбора режима IP_adapter
+    strength_ip=0.5,         # Коэфф. зашумления IP_adapter
+    ip_adapter_strength=0.5, # Сила влияния IP_adapter
     controlnet_conditioning_scale=0.5, # Сила влияния ControlNet
-    ip_source_image=None,   # Исходное изображение IP_adapter
-    ip_adapter_image=None,  # Контрольное изображение IP_adapter
-    remove_bg=None,         # Удаление фона с изображения
-    LCM_adapter=None,       # Параметр для выбора LCM_adapter
+    ip_source_image=None,    # Исходное изображение IP_adapter
+    ip_adapter_image=None,   # Контрольное изображение IP_adapter
+    remove_bg=None,          # Удаление фона с изображения
+    use_LCM_adapter=False,   # Параметр для включения LCM_adapter
+    LCM_adapter=None,        # Параметр для выбора типа LCM_adapter
+    use_DDIMScheduler=False, # Параметр для включения DDIMScheduler
     progress=gr.Progress(track_tqdm=True)
 ):
 
@@ -494,9 +496,9 @@ def infer(
         else:
             # Генерация изображений с LCM_Adapter ---------------------------------------------------------------------------------------------
             
-            if use_LCM_adapter == True:
+            if use_LCM_adapter:
 
-                print('use_LCM_adapter = True')
+                print('use_LCM_adapter = ', use_LCM_adapter)
                 
                 if LCM_adapter == "lcm-lora-sdv1-5":
                     adapter_id = "latent-consistency/lcm-lora-sdv1-5"
@@ -529,9 +531,9 @@ def infer(
             else:        
                 # Генерация изображений с DDIMScheduler ---------------------------------------------------------------------------------------------
 
-                if use_DDIMScheduler == True:
+                if use_DDIMScheduler:
 
-                    print('use_DDIMScheduler = True')
+                    print('use_DDIMScheduler = ', use_DDIMScheduler)
 
                     generator = torch.Generator(device).manual_seed(seed)
 
@@ -887,7 +889,9 @@ with gr.Blocks(css=css) as demo:
             ip_source_image,    # Исходное изображение IP_adapter
             ip_adapter_image,   # Контрольное изображение IP_adapter
             remove_bg,          # Удаление фона с изображения
-            LCM_adapter,        # Параметр для выбора LCM_adapter
+            use_LCM_adapter,    # Параметр для включения LCM_adapter
+            LCM_adapter,        # Параметр для выбора типа LCM_adapter
+            use_DDIMScheduler,  # Параметр для включения DDIMScheduler
         ],
         outputs=[result],
     )