From ffb9aac8266a677cf2c2e30d64b43b87bf00a3eb Mon Sep 17 00:00:00 2001 From: kijai <40791699+kijai@users.noreply.github.com> Date: Sun, 1 Sep 2024 19:54:30 +0300 Subject: [PATCH] Update fp8_optimization.py --- fp8_optimization.py | 1 - 1 file changed, 1 deletion(-) diff --git a/fp8_optimization.py b/fp8_optimization.py index 3f7aa9c..1e7c42b 100644 --- a/fp8_optimization.py +++ b/fp8_optimization.py @@ -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