Spaces:
Paused
Paused
File size: 520 Bytes
59d751c |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 |
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
|