Add format metadata to CLIP save to make compatible with diffusers safetensors loading (#4233)

This commit is contained in:
Silver 2024-08-06 07:45:24 +02:00 committed by GitHub
parent 841e74ac40
commit f3bc40223a
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 1 additions and 0 deletions

View File

@ -264,6 +264,7 @@ class CLIPSave:
metadata = {}
if not args.disable_metadata:
metadata["format"] = "pt"
metadata["prompt"] = prompt_info
if extra_pnginfo is not None:
for x in extra_pnginfo: