model_index.json

{
  "transformers_version": "4.22.0",
  "model_name": "DDPM-test",
  "architecture": "UNet",
  "config": {
    "T": 1000,
    "ch": 128,
    "ch_mult": [1, 2, 3, 4],
    "attn": [2],
    "num_res_blocks": 2,
    "dropout": 0.15
  },
  "framework": "pytorch",
  "file_format": "pt",
  "training_config": {
    "epoch": 200,
    "batch_size": 80,
    "lr": 0.0001,
    "multiplier": 2.0,
    "beta_1": 0.0001,
    "beta_T": 0.02,
    "img_size": 32,
    "grad_clip": 1.0,
    "device": "cuda:0"
  },
  "evaluation_config": {
    "test_load_weight": "ckpt_199_.pt",
    "sampled_dir": "./SampledImgs/",
    "sampledNoisyImgName": "NoisyNoGuidenceImgs.png",
    "sampledImgName": "SampledNoGuidenceImgs.png",
    "nrow": 8
  },
  "_class_name": ["diffusers.DiffusionPipeline", "DiffusionPipeline"]
}