Update fp8_optimization.py

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

View File

@ -1,3 +1,5 @@
#based on ComfyUI's and MinusZoneAI's fp8_linear optimization
import torch
import torch.nn as nn
from types import MethodType