Fix dtype parameter: use dtype instead of torch_dtype, remove device_map

acestep/handler.py

@@ -492,9 +492,8 @@ class AceStepHandler:
                             acestep_v15_checkpoint_path,
                             trust_remote_code=True,
                             attn_implementation=candidate,
-                            torch_dtype=self.dtype,
-                            device_map=None,  # Disable automatic device mapping
-                            low_cpu_mem_usage=False,  # Disable meta device for custom model with assertions
+                            dtype=self.dtype,
+                            low_cpu_mem_usage=False,  # Force immediate weight loading
                         )
                         attn_implementation = candidate
                         break