Update fp8_optimization.py

2024-09-01 19:54:30 +03:00
parent 29a5aa825d
commit ffb9aac826
1 changed files with 0 additions and 1 deletions
@@ -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