Fix fp8 fast lora offload.

This commit is contained in:
comfyanonymous 2025-12-04 19:09:35 -05:00 committed by GitHub
parent 83a0fa107b
commit c217243b56
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -449,7 +449,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: if len(self.weight_function) == 0 and len(self.bias_function) == 0:
try: try:
out = fp8_linear(self, input) out = fp8_linear(self, input)
if out is not None: if out is not None: