Update fp8_optimization.py

This commit is contained in:
kijai 2024-09-01 19:54:30 +03:00
parent 29a5aa825d
commit ffb9aac826

View File

@ -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