From 8bc53a6e71372fb5b0e41eb92e1688d31a98a56b Mon Sep 17 00:00:00 2001 From: "Lt.Dr.Data" Date: Tue, 16 May 2023 18:14:35 +0900 Subject: [PATCH] fix --- nodes.py | 11 ++++------- 1 file changed, 4 insertions(+), 7 deletions(-) diff --git a/nodes.py b/nodes.py index b8c18b9ad..bf3304aa9 100644 --- a/nodes.py +++ b/nodes.py @@ -305,7 +305,7 @@ class SaveLatent: metadata[x] = json.dumps(extra_pnginfo[x]) file = f"{filename}_{counter:05}_.latent" - sft.save_file(samples, file, metadata=metadata) + sft.save_file(samples, os.path.join(full_output_folder, file), metadata=metadata) return {} @@ -326,12 +326,9 @@ class LoadLatent: def load(self, latent): file = folder_paths.get_annotated_filepath(latent, self.input_dir) - samples = sft.load_file(file, device="cpu") - print(samples.keys()) - - # print(f"latent meta: {loaded_metadata}") - - return (samples, ) + latent = sft.load_file(file, device="cpu") + + return (latent, ) class CheckpointLoader: