[pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci

Author: pre-commit-ci[bot]

src/animatediff/models/motion_module.py

@@ -128,9 +128,9 @@ def forward(
         encoder_hidden_states: Optional[Tensor] = None,
         attention_mask: Optional[Tensor] = None,
     ):
-        assert (
-            hidden_states.dim() == 5
-        ), f"Expected hidden_states to have ndim=5, but got ndim={hidden_states.dim()}."
+        assert hidden_states.dim() == 5, (
+            f"Expected hidden_states to have ndim=5, but got ndim={hidden_states.dim()}."
+        )
         video_length = hidden_states.shape[2]
         hidden_states = rearrange(hidden_states, "b c f h w -> (b f) c h w")
 

src/animatediff/settings.py

@@ -43,11 +43,11 @@ def __call__(self, settings: BaseSettings) -> Dict[str, Any]:  # noqa C901
         merged_config = dict()  # create an empty dict to merge configs into
         for idx, path in enumerate(self.json_config_path):
             if path.exists() and path.is_file():  # check if the path exists and is a file
-                logger.debug(f"{classname}: loading config #{idx+1} from {path}")
+                logger.debug(f"{classname}: loading config #{idx + 1} from {path}")
                 merged_config.update(json.loads(path.read_text(encoding=encoding)))
-                logger.debug(f"{classname}: config state #{idx+1}: {merged_config}")
+                logger.debug(f"{classname}: config state #{idx + 1}: {merged_config}")
             else:
-                raise FileNotFoundError(f"{classname}: config #{idx+1} at {path} not found or not a file")
+                raise FileNotFoundError(f"{classname}: config #{idx + 1} at {path} not found or not a file")
 
         logger.debug(f"{classname}: loaded config: {merged_config}")
         return merged_config  # return the merged config

src/animatediff/utils/device.py

@@ -11,7 +11,7 @@
 def device_info_str(device: torch.device) -> str:
     device_info = torch.cuda.get_device_properties(device)
     return (
-        f"{device_info.name} {ceil(device_info.total_memory / 1024 ** 3)}GB, "
+        f"{device_info.name} {ceil(device_info.total_memory / 1024**3)}GB, "
         + f"CC {device_info.major}.{device_info.minor}, {device_info.multi_processor_count} SM(s)"
     )
 

src/animatediff/utils/pipeline.py

@@ -19,7 +19,7 @@ def send_to_device(
     force_half: bool = False,
     compile: bool = False,
 ) -> AnimationPipeline:
-    logger.info(f"Sending pipeline to device \"{device.type}{device.index if device.index else ''}\"")
+    logger.info(f'Sending pipeline to device "{device.type}{device.index if device.index else ""}"')
 
     # Freeze model weights and force-disable training
     if freeze or compile: