wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-07-05 22:21:31 +08:00
Code Issues Actions 17 Packages Projects Releases Wiki Activity

Support mixed precision ops

Browse Source
This commit is contained in:
kijai 2026-06-30 18:12:42 +03:00
parent c4059e483e
commit 4eb48f92bb
1 changed files with 7 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
comfy_extras/nodes_sam3d_body.py
View File

@ -69,7 +69,13 @@ class SAM3DBody_Loader(io.ComfyNode):
device=load_device, model_params=-1, weight_dtype=weight_dtype, device=load_device, model_params=-1, weight_dtype=weight_dtype,
) )
manual_cast_dtype = comfy.model_management.unet_manual_cast(torch_dtype, load_device) manual_cast_dtype = comfy.model_management.unet_manual_cast(torch_dtype, load_device)
operations = comfy.ops.pick_operations(torch_dtype, manual_cast_dtype, load_device=load_device, disable_fast_fp8=True)
quant_config = comfy.utils.detect_layer_quantization(sd, "")
if quant_config is not None:
operations = comfy.ops.mixed_precision_ops(quant_config, torch_dtype)
logging.info("SAM3D-Body: detected mixed precision quantization, using MixedPrecisionOps")
else:
operations = comfy.ops.pick_operations(torch_dtype, manual_cast_dtype, load_device=load_device, disable_fast_fp8=True)
model = SAM3DBody(dtype=torch_dtype, operations=operations) model = SAM3DBody(dtype=torch_dtype, operations=operations)
model.load_state_dict(sd, strict=False) model.load_state_dict(sd, strict=False)