xinyun/ComfyUI-CogVideoXWrapper
1
0
Fork 0
mirror of https://git.datalinker.icu/kijai/ComfyUI-CogVideoXWrapper.git synced 2026-01-23 18:24:23 +08:00
Code Issues Packages Projects Releases Wiki Activity

Update fp8_optimization.py

Browse Source
This commit is contained in:
kijai 2024-09-01 19:54:30 +03:00
parent 29a5aa825d
commit ffb9aac826
1 changed files with 0 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
fp8_optimization.py
View File

@ -2,7 +2,6 @@
import torch import torch
import torch.nn as nn import torch.nn as nn
from types import MethodType
def fp8_linear_forward(cls, original_dtype, input): def fp8_linear_forward(cls, original_dtype, input):
weight_dtype = cls.weight.dtype weight_dtype = cls.weight.dtype