github/ai-toolkit
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

Fixed saving full model of z_image l2p for finetuning

Browse Source
This commit is contained in:
Jaret Burkett
2026-05-23 07:36:24 -06:00
parent 6102370df9
commit c6ecc03ccd
1 changed files with 0 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
extensions_built_in/diffusion_models/z_image/z_image_l2p_model.py
View File

@@ -557,7 +557,3 @@ class ZImageL2PModel(ZImageModel):
continue continue
save_dict[key] = value.to("cpu").to(save_dtype) save_dict[key] = value.to("cpu").to(save_dtype)
save_file(save_dict, output_path, metadata=meta) save_file(save_dict, output_path, metadata=meta)
meta_path = output_path + ".aitk_meta.yaml"
with open(meta_path, "w") as f:
yaml.dump(meta, f)