wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-01-10 06:10:50 +08:00
Code Issues Actions 10 Packages Projects Releases Wiki Activity

Update ops.py

Browse Source
This commit is contained in:
comfyanonymous 2025-12-12 11:37:32 -05:00 committed by GitHub
parent 97eeb4d1c6
commit e140d3d496
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
comfy/ops.py
View File

@ -534,7 +534,7 @@ def mixed_precision_ops(quant_config={}, compute_dtype=torch.bfloat16, full_prec
if dtype != MixedPrecisionOps._compute_dtype: if dtype != MixedPrecisionOps._compute_dtype:
self.comfy_cast_weights = True self.comfy_cast_weights = True
if self._has_bias: if self._has_bias:
self.bias = torch.nn.Parameter(torch.empty(out_features, device=device, dtype=dtype)) self.bias = torch.nn.Parameter(torch.empty(self.out_features, device=device, dtype=dtype))
else: else:
self.register_parameter("bias", None) self.register_parameter("bias", None)
else: else:
@ -567,7 +567,7 @@ def mixed_precision_ops(quant_config={}, compute_dtype=torch.bfloat16, full_prec
) )
if self._has_bias: if self._has_bias:
self.bias = torch.nn.Parameter(torch.empty(out_features, device=device, dtype=MixedPrecisionOps._compute_dtype)) self.bias = torch.nn.Parameter(torch.empty(self.out_features, device=device, dtype=MixedPrecisionOps._compute_dtype))
else: else:
self.register_parameter("bias", None) self.register_parameter("bias", None)