Update asymmetric-tiling-sd-webui-3.0/scripts/asymmetric_tiling_UNIFIED.py

asymmetric-tiling-sd-webui-3.0/scripts/asymmetric_tiling_UNIFIED.py

@@ -1793,57 +1793,61 @@ def custom_padding_forward(input_tensor, weight, bias, stride, padding, dilation
         else:
             req_pad_h = req_pad_w = 0
         
-        # ===== ШАГ 5: Применяем тайлинг-специфичный padding =====
-        # (Здесь должна быть оригинальная логика из custom_padding_forward)
-        # Для примера упрощенная версия:
-        
-        mode_x = params.get('mode_x', 'Default (Off)')
-        mode_y = params.get('mode_y', 'Default (Off)')
-        
-        # Проверяем zoom mode
-        if params.get('use_zoom', False):
-            # Применяем zoom (упрощенно - нужна полная реализация)
-            x = F.pad(input_tensor, (req_pad_w, req_pad_w, req_pad_h, req_pad_h), mode='replicate')
-        else:
-            # Стандартные режимы
-            x = input_tensor
-            
-            # Ось Y
-            if mode_y == 'Circular':
-                x = F.pad(x, (0, 0, req_pad_h, req_pad_h), mode='circular')
-            elif mode_y == 'Mirror (Reflect)':
-                x = F.pad(x, (0, 0, req_pad_h, req_pad_h), mode='reflect')
-            else:
-                x = F.pad(x, (0, 0, req_pad_h, req_pad_h), mode='constant', value=0)
-            
-            # Ось X
-            if mode_x == 'Circular':
-                x = F.pad(x, (req_pad_w, req_pad_w, 0, 0), mode='circular')
-            elif mode_x == 'Mirror (Reflect)':
-                x = F.pad(x, (req_pad_w, req_pad_w, 0, 0), mode='reflect')
-            else:
-                x = F.pad(x, (req_pad_w, req_pad_w, 0, 0), mode='constant', value=0)
-        
-        # ===== ШАГ 6: Применяем свертку с нулевым padding (т.к. уже добавили) =====
-        output = F.conv2d(x, weight, bias, stride, 0, dilation, groups)
+        # FIX V3.7: Не выходим рано, если включен ZOOM!
+        if req_pad_h == 0 and req_pad_w == 0 and not params.get('use_zoom', False):
+            return F.conv2d(input_tensor, weight, bias, stride, padding, dilation, groups)
         
-        # ===== ШАГ 7: КРИТИЧЕСКОЕ ИСПРАВЛЕНИЕ - Корректируем размеры выхода =====
-        output = ensure_output_size_match(output, expected_out_h, expected_out_w)
+        mode_x = params.get('mode_x', MODE_OFF)
+        mode_y = params.get('mode_y', MODE_OFF)
         
-        return output
-    
-    except Exception as e:
-        print(f"[Advanced Tiling] Error in custom_padding_forward_FIXED: {e}")
-        import traceback
-        traceback.print_exc()
-        # Fallback к стандартной свертке
-        return F.conv2d(input_tensor, weight, bias, stride, padding, dilation, groups)
-
         # ═══════════════════════════════════════════════════════════════════
-        # СПЕЦИАЛЬНЫЕ РЕЖИМЫ (Cubemap, Panorama) - используют свои функции
+        # ZOOM SYSTEM - обрабатываем первым, т.к. может комбинироваться с другими режимами
         # ═══════════════════════════════════════════════════════════════════
+        if params.get('use_zoom', False):
+            try:
+                from advanced_zoom_extension import apply_unified_zoom, validate_zoom_params
+                
+                raw_zoom_params = {
+                    'zoom_factor': params.get('zoom_factor', 0.0),
+                    'zoom_mode': params.get('zoom_mode', ZoomMode.UNIFORM),
+                    'blend_mode': params.get('zoom_blend_mode', BlendMode.SMOOTH),
+                    'center_x': params.get('zoom_center_x', 0.5),
+                    'center_y': params.get('zoom_center_y', 0.5),
+                    'aspect_ratio_x': params.get('zoom_aspect_ratio_x', 1.0),
+                    'aspect_ratio_y': params.get('zoom_aspect_ratio_y', 1.0),
+                    'rotation_degrees': params.get('zoom_rotation_degrees', 0.0),
+                    'perspective_strength': params.get('zoom_perspective_strength', 0.0),
+                    'perspective_angle_x': params.get('zoom_perspective_angle_x', 0.0),
+                    'perspective_angle_y': params.get('zoom_perspective_angle_y', 0.0),
+                    'blend_width_ratio': params.get('zoom_blend_width_ratio', 0.1),
+                    'blend_strength': params.get('zoom_blend_strength', 1.0),
+                    'warp_strength': params.get('zoom_warp_strength', 0.0),
+                    'warp_frequency': params.get('zoom_warp_frequency', 2.0),
+                    'use_adaptive_blend': params.get('zoom_use_adaptive_blend', False),
+                    'edge_threshold': params.get('zoom_edge_threshold', 0.1),
+                }
+                
+                zoom_params = validate_zoom_params(raw_zoom_params)
+                
+                x = apply_unified_zoom(
+                    input_tensor, 
+                    pad_h=req_pad_h, 
+                    pad_w=req_pad_w,
+                    **zoom_params
+                )
+                
+                output = F.conv2d(x, weight, bias, stride, (0, 0), dilation, groups)
+                
+                # ✅ КРИТИЧЕСКОЕ ИСПРАВЛЕНИЕ: корректируем размеры
+                output = ensure_output_size_match(output, expected_out_h, expected_out_w)
+                return output
+            
+            except Exception as e:
+                print(f"⚠ Zoom Error: {e}")
+                import traceback
+                traceback.print_exc()
+                return F.conv2d(input_tensor, weight, bias, stride, padding, dilation, groups)
         
-        # ────────── CUBEMAP ──────────
         if mode_x == MODE_CUBEMAP or mode_y == MODE_CUBEMAP:
             cubemap_engine = params.get('cubemap_engine', 'A (Fast)')
             cubemap_pad_mode = params.get('cubemap_pad_mode', 'replicate')
@@ -1916,6 +1920,8 @@ def custom_padding_forward(input_tensor, weight, bias, stride, padding, dilation
                     enabled=True
                 )
 
+            # ✅ КРИТИЧЕСКОЕ ИСПРАВЛЕНИЕ: корректируем размеры
+            out_cube = ensure_output_size_match(out_cube, expected_out_h, expected_out_w)
             return out_cube
         
         # ────────── PANORAMA ──────────
@@ -1985,6 +1991,8 @@ def custom_padding_forward(input_tensor, weight, bias, stride, padding, dilation
                         enabled=True
                     )
 
+                # ✅ КРИТИЧЕСКОЕ ИСПРАВЛЕНИЕ: корректируем размеры
+                out_pano = ensure_output_size_match(out_pano, expected_out_h, expected_out_w)
                 return out_pano
 
             # Engine B: fast pole-correct padding
@@ -2027,6 +2035,8 @@ def custom_padding_forward(input_tensor, weight, bias, stride, padding, dilation
                         enabled=True
                     )
                 
+                # ✅ КРИТИЧЕСКОЕ ИСПРАВЛЕНИЕ: корректируем размеры
+                out_pano = ensure_output_size_match(out_pano, expected_out_h, expected_out_w)
                 return out_pano
 
             # Engine A (legacy): simple circular wrap
@@ -2057,6 +2067,8 @@ def custom_padding_forward(input_tensor, weight, bias, stride, padding, dilation
                         enabled=True
                     )
                 
+                # ✅ КРИТИЧЕСКОЕ ИСПРАВЛЕНИЕ: корректируем размеры
+                out_pano = ensure_output_size_match(out_pano, expected_out_h, expected_out_w)
                 return out_pano
 
         # ═══════════════════════════════════════════════════════════════════
@@ -2167,8 +2179,12 @@ def custom_padding_forward(input_tensor, weight, bias, stride, padding, dilation
                 enabled=True
             )
             
+            # ✅ КРИТИЧЕСКОЕ ИСПРАВЛЕНИЕ: корректируем размеры
+            out_final = ensure_output_size_match(out_final, expected_out_h, expected_out_w)
             return out_final  # ← ИСПРАВЛЕНО: возвращаем с multires!
         else:
+            # ✅ КРИТИЧЕСКОЕ ИСПРАВЛЕНИЕ: корректируем размеры
+            out_advanced = ensure_output_size_match(out_advanced, expected_out_h, expected_out_w)
             return out_advanced  # ← Без multires
 
     except Exception as e: