Allegro / model_index.json
LarryTsai's picture
Add `model_index.json` and update modeling configs (#2)
e3d6332 verified
raw
history blame
414 Bytes
{
"_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"
]
}