fix(mm): use correct import path for ModelMixin

This commit is contained in:
psychedelicious 2024-03-01 13:05:16 +11:00
parent b880a31039
commit 38474c9797

View File

@ -25,7 +25,7 @@ from enum import Enum
from typing import Literal, Optional, Type, Union
import torch
from diffusers import ModelMixin
from diffusers.models.modeling_utils import ModelMixin
from pydantic import BaseModel, ConfigDict, Discriminator, Field, Tag, TypeAdapter
from typing_extensions import Annotated, Any, Dict