|
--- |
|
license: creativeml-openrail-m |
|
--- |
|
|
|
### MODEL_CONFIG_DDIM_TRAIN |
|
|
|
``` |
|
MODEL_CONFIG_DDIM = { |
|
"_class_name": "StableDiffusionPipeline", |
|
"_diffusers_version": "0.6.0", |
|
"scheduler": [ |
|
"diffusers", |
|
"DDIMScheduler" |
|
], |
|
"text_encoder": [ |
|
"transformers", |
|
"CLIPTextModel" |
|
], |
|
"tokenizer": [ |
|
"transformers", |
|
"CLIPTokenizer" |
|
], |
|
"unet": [ |
|
"diffusers", |
|
"UNet2DConditionModel" |
|
], |
|
"vae": [ |
|
"diffusers", |
|
"AutoencoderKL" |
|
] |
|
} |
|
``` |
|
|
|
### MODEL_CONFIG_DDIM_SAVE |
|
``` |
|
MODEL_CONFIG_DDIM_SAVE = { |
|
"_class_name": "StableDiffusionPipeline", |
|
"_diffusers_version": "0.9.0.dev0", |
|
"scheduler": [ |
|
"diffusers", |
|
"DDIMScheduler" |
|
], |
|
"text_encoder": [ |
|
"transformers", |
|
"CLIPTextModel" |
|
], |
|
"tokenizer": [ |
|
"transformers", |
|
"CLIPTokenizer" |
|
], |
|
"unet": [ |
|
"diffusers", |
|
"UNet2DConditionModel" |
|
], |
|
"vae": [ |
|
"diffusers", |
|
"AutoencoderKL" |
|
] |
|
} |
|
``` |
|
|
|
|
|
### SCHEDULER_CONFIG_DDIM_TRAIN |
|
``` |
|
SCHEDULER_CONFIG_DDIM_TRAIN = { |
|
"_class_name": "DDIMScheduler", |
|
"_diffusers_version": "0.6.0", |
|
"beta_end": 0.012, |
|
"beta_schedule": "scaled_linear", |
|
"beta_start": 0.00085, |
|
"clip_sample": false, |
|
"num_train_timesteps": 1000, |
|
"set_alpha_to_one": false, |
|
"skip_prk_steps": true, |
|
"steps_offset": 1, |
|
"trained_betas": null |
|
|
|
} |
|
|
|
``` |
|
### SCHEDULER_CONFIG_DDIM_SAVE |
|
``` |
|
SCHEDULER_CONFIG_DDIM_SAVE = { |
|
"_class_name": "DDIMScheduler", |
|
"_diffusers_version": "0.9.0.dev0", |
|
"beta_end": 0.012, |
|
"beta_schedule": "scaled_linear", |
|
"beta_start": 0.00085, |
|
"clip_sample": false, |
|
"num_train_timesteps": 1000, |
|
"prediction_type": "epsilon", |
|
"set_alpha_to_one": false, |
|
"skip_prk_steps": true, |
|
"steps_offset": 1, |
|
"trained_betas": null |
|
} |
|
``` |