imjunaidafzal/stable-diffusion-for-finetune

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
}