| | --- |
| | 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 |
| | } |
| | ``` |