Fixing class level import

This commit is contained in:
Martin Kristiansen 2023-08-17 19:33:54 -04:00 committed by psychedelicious
parent 4f1008f31f
commit 33ee418d8c

View File

@ -307,7 +307,8 @@ class TextualInversionManager(BaseTextualInversionManager):
class ONNXModelPatcher:
from .models.base import IAIOnnxRuntimeModel, OnnxRuntimeModel
from .models.base import IAIOnnxRuntimeModel
from diffusers import OnnxRuntimeModel
@classmethod
@contextmanager