Upload config.py

config.py

@@ -370,18 +370,18 @@ class DiffusionPipelineConfig:
         print(">>> PIPELINE TYPE:", type(pipeline))
     
         # Try to move each component using .to_empty()
-        for name in ["unet", "transformer", "vae", "text_encoder"]:
-            module = getattr(pipeline, name, None)
-            if isinstance(module, torch.nn.Module):
-                try:
-                    print(f">>> Moving {name} to {device} using to_empty()")
-                    module.to_empty(device)
-                except Exception as e:
-                    print(f">>> WARNING: {name}.to_empty({device}) failed: {e}")
-                    try:
-                        print(f">>> Falling back to {name}.to({device})")
-                        module.to(device)
-                    except Exception as ee:
+        for name in ["unet", "transformer", "vae", "text_encoder"]:  
+            module = getattr(pipeline, name, None)  
+            if isinstance(module, torch.nn.Module):  
+                try:  
+                    print(f">>> Moving {name} to {device} using to_empty()")  
+                    module.to_empty(device=device)  # Use keyword argument  
+                except Exception as e:  
+                    print(f">>> WARNING: {name}.to_empty({device}) failed: {e}")  
+                    try:  
+                        print(f">>> Falling back to {name}.to({device})")  
+                        module.to(device)  
+                    except Exception as ee:  
                         print(f">>> ERROR: {name}.to({device}) also failed: {ee}")
     
         # Identify main model (for patching)