Support loading fp16 latent files.

2023-05-19 02:15:32 -04:00 · 2023-05-19 02:15:32 -04:00 · 8bbd9815a9
parent 62a371e12b
commit 8bbd9815a9
1 changed files with 1 additions and 1 deletions
--- 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