AI/diffusers
1
0
Fork 0
mirror of https://github.com/huggingface/diffusers.git synced 2026-01-27 17:22:53 +03:00
Code Activity

Fix setting fp16 dtype in AnimateDiff convert script. (#7127)

Browse Source 
* update

* update
This commit is contained in:
Dhruv Nair
2024-02-29 22:47:39 +05:30
committed by GitHub
parent fcfa270fbd
commit 8f2d13c684
1 changed files with 4 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
scripts/convert_animatediff_motion_module_to_diffusers.py
View File

@@ -30,6 +30,7 @@ def get_args():
parser.add_argument("--output_path", type=str, required=True)
parser.add_argument("--use_motion_mid_block", action="store_true")
parser.add_argument("--motion_max_seq_length", type=int, default=32)
parser.add_argument("--save_fp16", action="store_true")
return parser.parse_args()
@@ -48,4 +49,6 @@ if __name__ == "__main__":
# skip loading position embeddings
adapter.load_state_dict(conv_state_dict, strict=False)
adapter.save_pretrained(args.output_path)
adapter.save_pretrained(args.output_path, variant="fp16", torch_dtype=torch.float16)
if args.save_fp16:
adapter.to(torch.float16).save_pretrained(args.output_path, variant="fp16")