10 lines
153 B
Python
10 lines
153 B
Python
from dataclasses import dataclass
|
|
|
|
from diffusers import AutoencoderKL
|
|
|
|
|
|
@dataclass
|
|
class AutoencoderModel:
|
|
autoencoder: AutoencoderKL
|
|
path: str
|