xinyun/ComfyUI
1
0
Fork 0
mirror of https://git.datalinker.icu/comfyanonymous/ComfyUI synced 2026-09-08 19:57:03 +08:00
Code Issues Packages Projects Releases Wiki Activity

Fix fp8 model training

Browse Source
This commit is contained in:
Kohaku-Blueleaf 2025-09-14 11:29:39 +08:00
parent eea608e48f
commit 4a976adcd7
1 changed files with 7 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
comfy/ops.py
View File

@ -336,6 +336,7 @@ class fp8_ops(manual_cast):
return None return None
def forward_comfy_cast_weights(self, input): def forward_comfy_cast_weights(self, input):
if not self.training:
try: try:
out = fp8_linear(self, input) out = fp8_linear(self, input)
if out is not None: if out is not None: