Support loading fp16 latent files.

8bbd9815 · comfyanonymous · 62a371e1 · 8bbd9815
Commit 8bbd9815 authored May 19, 2023 by comfyanonymous
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

nodes.py nodes.py +1 -1

No files found.
--- a/nodes.py
+++ b/nodes.py
@@ -305,7 +305,7 @@ class LoadLatent:
    def load(self, latent):
        latent_path = folder_paths.get_annotated_filepath(latent)
        latent = safetensors.torch.load_file(latent_path, device="cpu")
-        samples = {"samples": latent["latent_tensor"]}
+        samples = {"samples": latent["latent_tensor"].float()}
        return (samples, )

    @classmethod