primerz commited on
Commit
1bcd4d0
·
verified ·
1 Parent(s): 5d00e28

Update models.py

Browse files
Files changed (1) hide show
  1. models.py +7 -0
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