| from typing import Any | |
| from pydantic import BaseModel | |
| class Components(BaseModel): | |
| # pipeline cls | |
| pipeline_cls: Any = None | |
| # Tokenizers | |
| tokenizer: Any = None | |
| tokenizer_2: Any = None | |
| tokenizer_3: Any = None | |
| # Text encoders | |
| text_encoder: Any = None | |
| text_encoder_2: Any = None | |
| text_encoder_3: Any = None | |
| # Autoencoder | |
| vae: Any = None | |
| # Denoiser | |
| transformer: Any = None | |
| controlnet: Any = None | |
| unet: Any = None | |
| # Scheduler | |
| scheduler: Any = None | |