handle weight casting in conversion script

a-r-r-o-w · a-r-r-o-w · commit 2a363a38b6de · 2024-12-12T11:01:25.000+01:00
diff --git a/scripts/convert_sana_to_diffusers.py b/scripts/convert_sana_to_diffusers.py
@@ -200,6 +200,7 @@ def main(args):
                 attrs=["bold"],
             )
         )
+        transformer = transformer.to(weight_dtype)
         transformer.save_pretrained(
             os.path.join(args.dump_path, "transformer"), safe_serialization=True, max_shard_size="5GB", variant=variant
         )

Original file line number	Diff line number	Diff line change
`@@ -200,6 +200,7 @@ def main(args):`
`200`	`200`	`attrs=["bold"],`
`201`	`201`	`)`
`202`	`202`	`)`
	`203`	`+ transformer = transformer.to(weight_dtype)`
`203`	`204`	`transformer.save_pretrained(`
`204`	`205`	`os.path.join(args.dump_path, "transformer"), safe_serialization=True, max_shard_size="5GB", variant=variant`
`205`	`206`	`)`