Spaces:
Runtime error
Runtime error
Update generator.py
Browse files- generator.py +0 -22
generator.py
CHANGED
|
@@ -175,28 +175,6 @@ class RetroArtConverter:
|
|
| 175 |
print(f"{model}: {status}")
|
| 176 |
print("===================\n")
|
| 177 |
|
| 178 |
-
print("=== UPGRADE VERIFICATION ===")
|
| 179 |
-
try:
|
| 180 |
-
from resampler_enhanced import EnhancedResampler
|
| 181 |
-
from ip_attention_processor_enhanced import EnhancedIPAttnProcessor2_0
|
| 182 |
-
|
| 183 |
-
resampler_check = isinstance(self.image_proj_model, EnhancedResampler) if hasattr(self, 'image_proj_model') and self.image_proj_model is not None else False
|
| 184 |
-
custom_attn_check = any(isinstance(p, EnhancedIPAttnProcessor2_0) for p in self.pipe.unet.attn_processors.values()) if hasattr(self, 'pipe') else False
|
| 185 |
-
|
| 186 |
-
print(f"Enhanced Perceiver Resampler: {'[OK] ACTIVE' if resampler_check else '[INFO] Not active'}")
|
| 187 |
-
print(f"Enhanced IP-Adapter Attention: {'[OK] ACTIVE' if custom_attn_check else '[INFO] Not active'}")
|
| 188 |
-
|
| 189 |
-
if resampler_check and custom_attn_check:
|
| 190 |
-
print("[SUCCESS] Face preservation upgrade fully active")
|
| 191 |
-
print(" Expected improvement: +10-15% face similarity")
|
| 192 |
-
elif resampler_check or custom_attn_check:
|
| 193 |
-
print("[PARTIAL] Some upgrades active")
|
| 194 |
-
else:
|
| 195 |
-
print("[INFO] Using standard components")
|
| 196 |
-
except Exception as e:
|
| 197 |
-
print(f"[INFO] Verification skipped: {e}")
|
| 198 |
-
print("============================\n")
|
| 199 |
-
|
| 200 |
|
| 201 |
def get_depth_map(self, image):
|
| 202 |
"""
|
|
|
|
| 175 |
print(f"{model}: {status}")
|
| 176 |
print("===================\n")
|
| 177 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 178 |
|
| 179 |
def get_depth_map(self, image):
|
| 180 |
"""
|