Skip to content

GitLab

Commit 8bbd9815 authored by comfyanonymous's avatar comfyanonymous
Browse files

Support loading fp16 latent files.

parent 62a371e1
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
nodes.py
View file @ 8bbd9815
...@@ -305,7 +305,7 @@ class LoadLatent: ...@@ -305,7 +305,7 @@ class LoadLatent:
def load(self, latent): def load(self, latent):
latent_path = folder_paths.get_annotated_filepath(latent) latent_path = folder_paths.get_annotated_filepath(latent)
latent = safetensors.torch.load_file(latent_path, device="cpu") latent = safetensors.torch.load_file(latent_path, device="cpu")
samples = {"samples": latent["latent_tensor"]} samples = {"samples": latent["latent_tensor"].float()}
return (samples, ) return (samples, )
@classmethod @classmethod
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment