wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-05-15 03:27:24 +08:00
Code Issues Actions 25 Packages Projects Releases Wiki Activity

fix

Browse Source
This commit is contained in:
Lt.Dr.Data 2023-05-16 18:21:24 +09:00
parent 8bc53a6e71
commit be3bea1eaf
1 changed files with 6 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
nodes.py
View File

@ -305,7 +305,10 @@ class SaveLatent:
metadata[x] = json.dumps(extra_pnginfo[x])
file = f"{filename}_{counter:05}_.latent"
sft.save_file(samples, os.path.join(full_output_folder, file), metadata=metadata)
file = os.path.join(full_output_folder, file)
print(file)
sft.save_file(samples, file, metadata=metadata)
return {}
@ -315,7 +318,7 @@ class LoadLatent:
@classmethod
def INPUT_TYPES(s):
files = [f for f in os.listdir(s.input_dir) if os.path.isfile(os.path.join(s.input_dir, f))]
files = [f for f in os.listdir(s.input_dir) if os.path.isfile(os.path.join(s.input_dir, f)) and f.endswith(".latent")]
return {"required": {"latent": [sorted(files), ]}, }
CATEGORY = "_for_testing"
@ -327,7 +330,7 @@ class LoadLatent:
file = folder_paths.get_annotated_filepath(latent, self.input_dir)
latent = sft.load_file(file, device="cpu")
return (latent, )