Upload app.py with huggingface_hub

app.py

@@ -22,10 +22,9 @@ from diffusers import (
     QwenImageImg2ImgPipeline
 )
 from diffusers.schedulers import (
-    DPMSolverMultistepScheduler,
+    FlowMatchEulerDiscreteScheduler,
     EulerDiscreteScheduler,
-    EulerAncestralDiscreteScheduler,
-    UniPCMultistepScheduler
+    FlowMatchHeunDiscreteScheduler
 )
 from huggingface_hub import hf_hub_download
 
@@ -59,84 +58,55 @@ if torch.cuda.is_available():
 # =================================================================
 
 # Конфигурации для разных типов schedulers
-# Основаны на исследовании популярных вариантов в Qwen-Image и сообществе
+# ⚠️ ВАЖНО: Qwen pipeline передает sigmas в set_timesteps()
+# Поэтому работают ТОЛЬКО schedulers которые поддерживают sigmas параметр!
 SCHEDULER_CONFIGS = {
-    # === ПОПУЛЯРНЫЕ (Рекомендуемые) ===
-    "DPM++ 2M Karras": {
-        "class": DPMSolverMultistepScheduler,
-        "config": {"algorithm_type": "dpmsolver++", "solver_order": 2, "use_karras_sigmas": True},
-        "alias": "dpmpp_2m_karras"
-    },
-    "DPM++ SDE Karras": {
-        "class": DPMSolverMultistepScheduler,
-        "config": {"algorithm_type": "sde-dpmsolver++", "solver_order": 2, "use_karras_sigmas": True},
-        "alias": "dpmpp_sde_karras"
-    },
-    "DPM++ 3M SDE Karras": {
-        "class": DPMSolverMultistepScheduler,
-        "config": {"algorithm_type": "sde-dpmsolver++", "solver_order": 3, "use_karras_sigmas": True},
-        "alias": "dpmpp_3m_sde_karras"
+    # === РЕКОМЕНДУЕМЫЕ (Совместимые с Qwen Flow Matching) ===
+    "Flow Match Euler": {
+        "class": FlowMatchEulerDiscreteScheduler,
+        "config": {},
+        "alias": "flow_euler"
     },
-    "Euler Ancestral": {
-        "class": EulerAncestralDiscreteScheduler,
+    "Flow Match Heun": {
+        "class": FlowMatchHeunDiscreteScheduler,
         "config": {},
-        "alias": "euler_a"
+        "alias": "flow_heun"
     },
     "Euler": {
         "class": EulerDiscreteScheduler,
         "config": {},
         "alias": "euler"
     },
-    
-    # === ЭКСПЕРИМЕНТАЛЬНЫЕ (Альтернативные варианты сигм) ===
-    "DPM++ 2M": {
-        "class": DPMSolverMultistepScheduler,
-        "config": {"algorithm_type": "dpmsolver++", "solver_order": 2},
-        "alias": "dpmpp_2m"
-    },
-    "DPM++ 2M Beta": {
-        "class": DPMSolverMultistepScheduler,
-        "config": {"algorithm_type": "dpmsolver++", "solver_order": 2, "use_beta_sigmas": True},
-        "alias": "dpmpp_2m_beta"
-    },
-    "DPM++ 2M Exponential": {
-        "class": DPMSolverMultistepScheduler,
-        "config": {"algorithm_type": "dpmsolver++", "solver_order": 2, "use_exponential_sigmas": True},
-        "alias": "dpmpp_2m_exp"
-    },
     "Euler Karras": {
         "class": EulerDiscreteScheduler,
         "config": {"use_karras_sigmas": True},
         "alias": "euler_karras"
     },
-    
-    # === БЫСТРЫЕ (Для экспериментов) ===
-    "UniPC": {
-        "class": UniPCMultistepScheduler,
-        "config": {},
-        "alias": "unipc"
+    "Euler Beta": {
+        "class": EulerDiscreteScheduler,
+        "config": {"use_beta_sigmas": True},
+        "alias": "euler_beta"
+    },
+    "Euler Exponential": {
+        "class": EulerDiscreteScheduler,
+        "config": {"use_exponential_sigmas": True},
+        "alias": "euler_exp"
     }
 }
 
-DEFAULT_SCHEDULER = "DPM++ 2M Karras"
+DEFAULT_SCHEDULER = "Flow Match Euler"
 
 # Для UI - группируем по категориям для удобства
 SCHEDULER_GROUPS = {
-    "⭐ Рекомендуемые": [
-        "DPM++ 2M Karras",
-        "DPM++ SDE Karras",
-        "DPM++ 3M SDE Karras",
-        "Euler Ancestral",
-        "Euler"
-    ],
-    "🧪 Экспериментальные": [
-        "DPM++ 2M",
-        "DPM++ 2M Beta",
-        "DPM++ 2M Exponential",
-        "Euler Karras"
+    "⭐ Flow Matching (Рекомендуемые)": [
+        "Flow Match Euler",
+        "Flow Match Heun"
     ],
-    "⚡ Быстрые": [
-        "UniPC"
+    "🔄 Euler Variants (Экспериментальные)": [
+        "Euler",
+        "Euler Karras",
+        "Euler Beta",
+        "Euler Exponential"
     ]
 }