Commented out debug code

This commit is contained in:
Jedrzej Kosinski
2024-06-13 00:10:18 -05:00
parent ee1ad61051
commit 05550a52b8
+4 -4
View File
@@ -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