nodes/HM-RunningHub-ComfyUI_RH_OminiControl
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Update rh_utils.py

Browse Source
This commit is contained in:
HM-RunningHub
2024-12-20 16:41:09 +08:00
committed by GitHub
parent d3c9982373
commit 1087735863
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
rh_utils.py
+2 -2
View File
@@ -48,11 +48,11 @@ def decode_latents(flux_dir, latents):
torch_dtype=torch.bfloat16,
).to("cuda")
latents = pipeline._unpack_latents(latents, g_width, g_height, pipeline.vae_scale_factor)
latents = pipeline._unpack_latents(latents, g_height, g_width, pipeline.vae_scale_factor)
latents = (
latents / pipeline.vae.config.scaling_factor
) + pipeline.vae.config.shift_factor
image = pipeline.vae.decode(latents, return_dict=False)[0]
image = pipeline.image_processor.postprocess(image, output_type="pil")
return FluxPipelineOutput(images=image)
return FluxPipelineOutput(images=image)