wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-03-27 03:53:27 +08:00
Code Issues Actions 21 Packages Projects Releases Wiki Activity

Fix sampling issue with fp16 intermediates. (#13099)

Browse Source
This commit is contained in:
comfyanonymous 2026-03-21 14:47:42 -07:00 committed by GitHub
parent a11f68dd3b
commit b5d32e6ad2
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
comfy/samplers.py
View File

@ -985,8 +985,8 @@ class CFGGuider:
self.inner_model, self.conds, self.loaded_models = comfy.sampler_helpers.prepare_sampling(self.model_patcher, noise.shape, self.conds, self.model_options) self.inner_model, self.conds, self.loaded_models = comfy.sampler_helpers.prepare_sampling(self.model_patcher, noise.shape, self.conds, self.model_options)
device = self.model_patcher.load_device device = self.model_patcher.load_device
noise = noise.to(device) noise = noise.to(device=device, dtype=torch.float32)
latent_image = latent_image.to(device) latent_image = latent_image.to(device=device, dtype=torch.float32)
sigmas = sigmas.to(device) sigmas = sigmas.to(device)
cast_to_load_options(self.model_options, device=device, dtype=self.model_patcher.model_dtype()) cast_to_load_options(self.model_options, device=device, dtype=self.model_patcher.model_dtype())
@ -1028,6 +1028,7 @@ class CFGGuider:
denoise_mask, _ = comfy.utils.pack_latents(denoise_masks) denoise_mask, _ = comfy.utils.pack_latents(denoise_masks)
else: else:
denoise_mask = denoise_masks[0] denoise_mask = denoise_masks[0]
denoise_mask = denoise_mask.float()
self.conds = {} self.conds = {}
for k in self.original_conds: for k in self.original_conds: