Apply suggestions from code review

Co-authored-by: Sayak Paul <[email protected]>

Author: SunMarc

src/diffusers/loaders/single_file_model.py

@@ -404,9 +404,7 @@ def from_single_file(cls, pretrained_model_link_or_path_or_dict: Optional[str] =
         model.eval()
 
         if device_map is not None:
-            device_map_kwargs = {
-                "device_map": device_map,
-            }
+            device_map_kwargs = {"device_map": device_map}
             dispatch_model(model, **device_map_kwargs)
 
         return model

src/diffusers/models/modeling_utils.py

@@ -235,13 +235,13 @@ def no_init_weights():
     def _skip_init(*args, **kwargs):
         pass
 
-    # # Save the original initialization functions
+    # Save the original initialization functions
     for name, init_func in TORCH_INIT_FUNCTIONS.items():
         setattr(torch.nn.init, name, _skip_init)
     try:
         yield
     finally:
-        # # Restore the original initialization functions
+        # Restore the original initialization functions
         for name, init_func in TORCH_INIT_FUNCTIONS.items():
             setattr(torch.nn.init, name, init_func)