mirrors/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2025-06-08 23:27:14 +00:00
Code Issues Packages Projects Releases Wiki Activity

Support loading fp16 latent files.

Browse Source
This commit is contained in:
comfyanonymous 2023-05-19 02:15:32 -04:00
parent 62a371e12b
commit 8bbd9815a9
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
nodes.py
View File

@ -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