mirror of
https://github.com/comfyanonymous/ComfyUI.git
synced 2025-12-20 19:42:59 +08:00
Fix fp8 fast lora offload.
This commit is contained in:
parent
83a0fa107b
commit
c217243b56
@ -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:
|
||||||
|
|||||||
Loading…
Reference in New Issue
Block a user