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