Upload app.py with huggingface_hub

app.py

@@ -21,7 +21,6 @@ from diffusers import (
     DiffusionPipeline,
     QwenImageImg2ImgPipeline,
     FlowMatchEulerDiscreteScheduler,
-    EulerDiscreteScheduler,
     FlowMatchHeunDiscreteScheduler
 )
 from huggingface_hub import hf_hub_download
@@ -56,55 +55,61 @@ if torch.cuda.is_available():
 # =================================================================
 
 # Конфигурации для разных типов schedulers
-# ⚠️ ВАЖНО: Qwen pipeline передает sigmas в set_timesteps()
-# Поэтому работают ТОЛЬКО schedulers которые поддерживают sigmas параметр!
+# ⚠️ ВАЖНО: Qwen использует Flow Matching, а не классическую diffusion!
+# Работают ТОЛЬКО Flow Matching schedulers с разными параметрами shift
 SCHEDULER_CONFIGS = {
-    # === РЕКОМЕНДУЕМЫЕ (Совместимые с Qwen Flow Matching) ===
-    "Flow Match Euler": {
+    # === Flow Matching Schedulers (единственные совместимые с Qwen) ===
+    "Flow Euler (Default)": {
         "class": FlowMatchEulerDiscreteScheduler,
         "config": {},
         "alias": "flow_euler"
     },
-    "Flow Match Heun": {
+    "Flow Euler (Shift 0.5)": {
+        "class": FlowMatchEulerDiscreteScheduler,
+        "config": {"shift": 0.5},
+        "alias": "flow_euler_shift05"
+    },
+    "Flow Euler (Shift 1.5)": {
+        "class": FlowMatchEulerDiscreteScheduler,
+        "config": {"shift": 1.5},
+        "alias": "flow_euler_shift15"
+    },
+    "Flow Euler (Shift 2.0)": {
+        "class": FlowMatchEulerDiscreteScheduler,
+        "config": {"shift": 2.0},
+        "alias": "flow_euler_shift20"
+    },
+    "Flow Heun (Default)": {
         "class": FlowMatchHeunDiscreteScheduler,
         "config": {},
         "alias": "flow_heun"
     },
-    "Euler": {
-        "class": EulerDiscreteScheduler,
-        "config": {},
-        "alias": "euler"
-    },
-    "Euler Karras": {
-        "class": EulerDiscreteScheduler,
-        "config": {"use_karras_sigmas": True},
-        "alias": "euler_karras"
-    },
-    "Euler Beta": {
-        "class": EulerDiscreteScheduler,
-        "config": {"use_beta_sigmas": True},
-        "alias": "euler_beta"
+    "Flow Heun (Shift 0.5)": {
+        "class": FlowMatchHeunDiscreteScheduler,
+        "config": {"shift": 0.5},
+        "alias": "flow_heun_shift05"
     },
-    "Euler Exponential": {
-        "class": EulerDiscreteScheduler,
-        "config": {"use_exponential_sigmas": True},
-        "alias": "euler_exp"
+    "Flow Heun (Shift 1.5)": {
+        "class": FlowMatchHeunDiscreteScheduler,
+        "config": {"shift": 1.5},
+        "alias": "flow_heun_shift15"
     }
 }
 
-DEFAULT_SCHEDULER = "Flow Match Euler"
+DEFAULT_SCHEDULER = "Flow Euler (Default)"
 
 # Для UI - группируем по категориям для удобства
 SCHEDULER_GROUPS = {
-    "⭐ Flow Matching (Рекомендуемые)": [
-        "Flow Match Euler",
-        "Flow Match Heun"
+    "🚀 Euler (Рекомендуемые)": [
+        "Flow Euler (Default)",
+        "Flow Euler (Shift 0.5)",
+        "Flow Euler (Shift 1.5)",
+        "Flow Euler (Shift 2.0)"
     ],
-    "🔄 Euler Variants (Экспериментальные)": [
-        "Euler",
-        "Euler Karras",
-        "Euler Beta",
-        "Euler Exponential"
+    "🎯 Heun (Более точные)": [
+        "Flow Heun (Default)",
+        "Flow Heun (Shift 0.5)",
+        "Flow Heun (Shift 1.5)"
     ]
 }