Fix issue when saving advanced prompt embeds. No such file or directory error
This commit is contained in:
@@ -1,3 +1,4 @@
|
|||||||
|
import os
|
||||||
from safetensors.torch import load_file, save_file
|
from safetensors.torch import load_file, save_file
|
||||||
|
|
||||||
|
|
||||||
@@ -115,6 +116,7 @@ class AdvancedPromptEmbeds:
|
|||||||
f"Cannot save key {key!r}: expected list of length 1, got {len(value)}"
|
f"Cannot save key {key!r}: expected list of length 1, got {len(value)}"
|
||||||
)
|
)
|
||||||
data[key] = value[0]
|
data[key] = value[0]
|
||||||
|
os.makedirs(os.path.dirname(path), exist_ok=True)
|
||||||
save_file(data, path, metadata=metadata)
|
save_file(data, path, metadata=metadata)
|
||||||
|
|
||||||
@classmethod
|
@classmethod
|
||||||
|
|||||||
Reference in New Issue
Block a user