We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 79ed752 commit 111f583Copy full SHA for 111f583
comfy/ops.py
@@ -336,9 +336,12 @@ def reset_parameters(self):
336
return None
337
338
def forward_comfy_cast_weights(self, input):
339
- out = fp8_linear(self, input)
340
- if out is not None:
341
- return out
+ try:
+ out = fp8_linear(self, input)
+ if out is not None:
342
+ return out
343
+ except Exception as e:
344
+ logging.info("Exception during fp8 op: {}".format(e))
345
346
weight, bias = cast_bias_weight(self, input)
347
return torch.nn.functional.linear(input, weight, bias)
0 commit comments