Spaces:
Runtime error
Runtime error
Update models.py
Browse files
models.py
CHANGED
|
@@ -134,6 +134,13 @@ def load_sdxl_pipeline(controlnets):
|
|
| 134 |
pipe.set_ip_adapter_scale(0.8)
|
| 135 |
print(" [OK] IP-Adapter loaded")
|
| 136 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 137 |
pipe = pipe.to(device)
|
| 138 |
|
| 139 |
# DEBUG: Check text_encoder type
|
|
|
|
| 134 |
pipe.set_ip_adapter_scale(0.8)
|
| 135 |
print(" [OK] IP-Adapter loaded")
|
| 136 |
|
| 137 |
+
# DEBUG: Check UNet configuration
|
| 138 |
+
print(f" [DEBUG] UNet cross_attention_dim: {pipe.unet.config.cross_attention_dim}")
|
| 139 |
+
if hasattr(pipe, 'image_proj_model'):
|
| 140 |
+
print(f" [DEBUG] Resampler output_dim: {pipe.image_proj_model.proj_out.out_features}")
|
| 141 |
+
else:
|
| 142 |
+
print(f" [DEBUG] WARNING: No image_proj_model found!")
|
| 143 |
+
|
| 144 |
pipe = pipe.to(device)
|
| 145 |
|
| 146 |
# DEBUG: Check text_encoder type
|