style: rename
This commit is contained in:
@@ -1,7 +1,7 @@
|
||||
from ._autoencoder import AutoencoderModel
|
||||
from ._pipeline import PipelineModel
|
||||
from ._autoencoder import Autoencoder
|
||||
from ._pipeline import Pipeline
|
||||
|
||||
__all__ = [
|
||||
"PipelineModel",
|
||||
"AutoencoderModel",
|
||||
"Pipeline",
|
||||
"Autoencoder",
|
||||
]
|
||||
|
||||
@@ -4,6 +4,6 @@ from diffusers import AutoencoderKL
|
||||
|
||||
|
||||
@dataclass
|
||||
class AutoencoderModel:
|
||||
class Autoencoder:
|
||||
autoencoder: AutoencoderKL
|
||||
path: str
|
||||
|
||||
@@ -4,6 +4,6 @@ from diffusers import StableDiffusionPipeline
|
||||
|
||||
|
||||
@dataclass
|
||||
class PipelineModel:
|
||||
class Pipeline:
|
||||
pipeline: StableDiffusionPipeline
|
||||
path: str
|
||||
|
||||
Reference in New Issue
Block a user