From 05550a52b8a98b7d3bbd26f79a486299cc50b9be Mon Sep 17 00:00:00 2001 From: Jedrzej Kosinski Date: Thu, 13 Jun 2024 00:10:18 -0500 Subject: [PATCH] Commented out debug code --- animatediff/motion_module_ad.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/animatediff/motion_module_ad.py b/animatediff/motion_module_ad.py index 4947019..4ee3b63 100644 --- a/animatediff/motion_module_ad.py +++ b/animatediff/motion_module_ad.py @@ -133,10 +133,10 @@ def has_img_encoder(mm_state_dict: dict[str, Tensor]): def normalize_ad_state_dict(mm_state_dict: dict[str, Tensor], mm_name: str) -> Tuple[dict[str, Tensor], AnimateDiffInfo]: - from pathlib import Path - with open(Path(__file__).parent.parent.parent / f"keys_{mm_name}.txt", "w") as afile: - for key, value in mm_state_dict.items(): - afile.write(f"{key}:\t{value.shape}\n") + # from pathlib import Path + # with open(Path(__file__).parent.parent.parent / f"keys_{mm_name}.txt", "w") as afile: + # for key, value in mm_state_dict.items(): + # afile.write(f"{key}:\t{value.shape}\n") # determine what SD model the motion module is intended for sd_type: str = None