Upload 2 files

generator.py

@@ -82,10 +82,8 @@ class RetroArtConverter:
             self.image_proj_model = None
         
         # Setup Compel
-        # TEMPORARILY DISABLED - SDXL token mismatch issue
         # Setup Compel
         self.compel, self.use_compel = setup_compel(self.pipe)
-        print("  [INFO] Compel temporarily disabled - using standard prompts")
         
         # Setup LCM scheduler
         setup_scheduler(self.pipe)

models.py

@@ -241,21 +241,22 @@ def setup_ip_adapter(pipe, image_encoder):
         # Load image_proj weights
         if image_proj_state_dict:
             try:
-                # Try strict loading first
-                image_proj_model.load_state_dict(image_proj_state_dict, strict=True)
-                print("  [OK] Resampler loaded with pretrained weights (strict)")
+                # Check if weights are nested under 'image_proj' key
+                if 'image_proj' in image_proj_state_dict and isinstance(image_proj_state_dict['image_proj'], dict):
+                    actual_weights = image_proj_state_dict['image_proj']
+                else:
+                    actual_weights = image_proj_state_dict
+                
+                # Try loading the weights
+                missing, unexpected = image_proj_model.load_state_dict(actual_weights, strict=False)
+                print("  [OK] Resampler loaded with pretrained weights")
+                if missing:
+                    print(f"    Missing keys: {missing[:5]}...")
+                if unexpected:
+                    print(f"    Unexpected keys: {unexpected[:5]}...")
             except Exception as e:
-                # Try non-strict if strict fails
-                try:
-                    missing, unexpected = image_proj_model.load_state_dict(image_proj_state_dict, strict=False)
-                    print(f"  [OK] Resampler loaded with pretrained weights (non-strict)")
-                    if missing:
-                        print(f"    Missing keys: {missing[:5]}...")  # Show first 5
-                    if unexpected:
-                        print(f"    Unexpected keys: {unexpected[:5]}...")  # Show first 5
-                except Exception as e2:
-                    print(f"  [WARNING] Could not load Resampler weights: {e2}")
-                    print("  Using randomly initialized Resampler")
+                print(f"  [WARNING] Could not load Resampler weights: {e}")
+                print("  Using randomly initialized Resampler")
         else:
             print("  [WARNING] No image_proj weights found in state dict")
             print("  Using randomly initialized Resampler")