Allegro/model_index.json
YuxuanCai e3d6332920
Add model_index.json and update modeling configs (#2)
- add model_index.json; update configs (43d06687e1b23afacab5980e452638293b3722b0)
- update (7defb981dbb797a900f485c84f483d6d21e9520a)


Co-authored-by: Aryan V S <a-r-r-o-w@users.noreply.huggingface.co>
2024-10-29 03:50:46 +00:00

25 lines
414 B
JSON

{
"_class_name": "AllegroPipeline",
"_diffusers_version": "0.31.0.dev0",
"scheduler": [
"diffusers",
"EulerAncestralDiscreteScheduler"
],
"text_encoder": [
"transformers",
"T5EncoderModel"
],
"tokenizer": [
"transformers",
"T5Tokenizer"
],
"transformer": [
"diffusers",
"AllegroTransformer3DModel"
],
"vae": [
"diffusers",
"AutoencoderKLAllegro"
]
}