mirror of
https://git.datalinker.icu/kijai/ComfyUI-CogVideoXWrapper.git
synced 2025-12-08 20:34:23 +08:00
Update fp8_optimization.py
This commit is contained in:
parent
29a5aa825d
commit
ffb9aac826
@ -2,7 +2,6 @@
|
||||
|
||||
import torch
|
||||
import torch.nn as nn
|
||||
from types import MethodType
|
||||
|
||||
def fp8_linear_forward(cls, original_dtype, input):
|
||||
weight_dtype = cls.weight.dtype
|
||||
|
||||
Loading…
x
Reference in New Issue
Block a user