Add files using upload-large-folder tool

README.md

@@ -0,0 +1,22 @@
+# cfm_corr_150_ss0.0_ep500_ema-50k-run2
+
+This repository contains model artifacts and configuration files from the CFM_CORR_EMA_50k experiment.
+
+## Contents
+
+This folder contains:
+- Model checkpoints and weights
+- Configuration files (JSON)
+- Scheduler and UNet components
+- Training results and metadata
+- Sample directories (excluding image files)
+
+## Experiment Details
+
+- Experiment: CFM_CORR_EMA_50k
+- Folder: cfm_corr_150_ss0.0_ep500_ema-50k-run2
+- Images excluded from upload
+
+## Usage
+
+This repository contains the necessary files to reproduce or analyze the model from this specific experimental run.

config.json

@@ -0,0 +1,46 @@
+{
+    "save_model_per_ep": 5,
+    "checkpoint_per_ep": 5,
+    "ep_model_dir": "epochs",
+    "output_dir": "fm_cifar10",
+    "ckpt_dir": "ckpt",
+    "ckpt_data_file": "data.ckpt",
+    "is_save_all_model_epochs": false,
+    "args_key": "args",
+    "default_key": "default",
+    "final_key": "final",
+    "config_file": "config.json",
+    "optim_name": "adamw",
+    "shed_name": "poly_decay",
+    "sched_num_warmup_steps": 45000,
+    "optim_weight_decay": 0.0,
+    "optim_beta_1": 0.9,
+    "optim_beta_2": 0.999,
+    "optim_epsilon": 1e-08,
+    "sched_lr_end": 1e-07,
+    "sched_power": 1.0,
+    "sched_t_max": 200,
+    "sched_last_epoch": -1,
+    "training_result_file": "training_result.json",
+    "check_finished_file": "finished.json",
+    "project": "CFM_CORR_EMA_50k",
+    "name_postfix": "ema-50k-run2",
+    "saving_strategy": "best_valid_loss",
+    "optim_lr": 0.0005,
+    "model_id": "google/ddpm-cifar10-32",
+    "train_dataset": "cifar10",
+    "traj_dataset": "data/traj_dataset/TRAJ-DS_PS-DM_DDPM_PSS-1000_SZ-50000",
+    "batch_size": 256,
+    "num_epochs": 500,
+    "num_train_timesteps": 1000,
+    "predictor_num_inference_steps": 1000,
+    "corrector_num_inference_steps": 100,
+    "sigma_min": 0.0,
+    "sigma_src": 0.0,
+    "corr_trained_timestep": 150,
+    "ds_size": 50000,
+    "seed": 44,
+    "device": "cuda:0",
+    "overwrite": false,
+    "weight_decay": 0.0
+}

epochs/ep10/model_index.json

@@ -0,0 +1,15 @@
+{
+  "_class_name": "DDPMCorrectorPipeline",
+  "_diffusers_version": "0.34.0",
+  "allowed_ctrl_timesteps": [
+    150
+  ],
+  "scheduler": [
+    "diffusers",
+    "FlowMatchEulerDiscreteScheduler"
+  ],
+  "unet": [
+    "diffusers",
+    "UNet2DModel"
+  ]
+}

epochs/ep115/model_index.json

@@ -0,0 +1,15 @@
+{
+  "_class_name": "DDPMCorrectorPipeline",
+  "_diffusers_version": "0.34.0",
+  "allowed_ctrl_timesteps": [
+    150
+  ],
+  "scheduler": [
+    "diffusers",
+    "FlowMatchEulerDiscreteScheduler"
+  ],
+  "unet": [
+    "diffusers",
+    "UNet2DModel"
+  ]
+}

epochs/ep130/model_index.json

@@ -0,0 +1,15 @@
+{
+  "_class_name": "DDPMCorrectorPipeline",
+  "_diffusers_version": "0.34.0",
+  "allowed_ctrl_timesteps": [
+    150
+  ],
+  "scheduler": [
+    "diffusers",
+    "FlowMatchEulerDiscreteScheduler"
+  ],
+  "unet": [
+    "diffusers",
+    "UNet2DModel"
+  ]
+}

epochs/ep135/model_index.json

@@ -0,0 +1,15 @@
+{
+  "_class_name": "DDPMCorrectorPipeline",
+  "_diffusers_version": "0.34.0",
+  "allowed_ctrl_timesteps": [
+    150
+  ],
+  "scheduler": [
+    "diffusers",
+    "FlowMatchEulerDiscreteScheduler"
+  ],
+  "unet": [
+    "diffusers",
+    "UNet2DModel"
+  ]
+}

epochs/ep140/model_index.json

@@ -0,0 +1,15 @@
+{
+  "_class_name": "DDPMCorrectorPipeline",
+  "_diffusers_version": "0.34.0",
+  "allowed_ctrl_timesteps": [
+    150
+  ],
+  "scheduler": [
+    "diffusers",
+    "FlowMatchEulerDiscreteScheduler"
+  ],
+  "unet": [
+    "diffusers",
+    "UNet2DModel"
+  ]
+}

epochs/ep15/model_index.json

@@ -0,0 +1,15 @@
+{
+  "_class_name": "DDPMCorrectorPipeline",
+  "_diffusers_version": "0.34.0",
+  "allowed_ctrl_timesteps": [
+    150
+  ],
+  "scheduler": [
+    "diffusers",
+    "FlowMatchEulerDiscreteScheduler"
+  ],
+  "unet": [
+    "diffusers",
+    "UNet2DModel"
+  ]
+}

epochs/ep155/model_index.json

@@ -0,0 +1,15 @@
+{
+  "_class_name": "DDPMCorrectorPipeline",
+  "_diffusers_version": "0.34.0",
+  "allowed_ctrl_timesteps": [
+    150
+  ],
+  "scheduler": [
+    "diffusers",
+    "FlowMatchEulerDiscreteScheduler"
+  ],
+  "unet": [
+    "diffusers",
+    "UNet2DModel"
+  ]
+}

epochs/ep160/model_index.json

@@ -0,0 +1,15 @@
+{
+  "_class_name": "DDPMCorrectorPipeline",
+  "_diffusers_version": "0.34.0",
+  "allowed_ctrl_timesteps": [
+    150
+  ],
+  "scheduler": [
+    "diffusers",
+    "FlowMatchEulerDiscreteScheduler"
+  ],
+  "unet": [
+    "diffusers",
+    "UNet2DModel"
+  ]
+}

epochs/ep170/model_index.json

@@ -0,0 +1,15 @@
+{
+  "_class_name": "DDPMCorrectorPipeline",
+  "_diffusers_version": "0.34.0",
+  "allowed_ctrl_timesteps": [
+    150
+  ],
+  "scheduler": [
+    "diffusers",
+    "FlowMatchEulerDiscreteScheduler"
+  ],
+  "unet": [
+    "diffusers",
+    "UNet2DModel"
+  ]
+}

epochs/ep20/model_index.json

@@ -0,0 +1,15 @@
+{
+  "_class_name": "DDPMCorrectorPipeline",
+  "_diffusers_version": "0.34.0",
+  "allowed_ctrl_timesteps": [
+    150
+  ],
+  "scheduler": [
+    "diffusers",
+    "FlowMatchEulerDiscreteScheduler"
+  ],
+  "unet": [
+    "diffusers",
+    "UNet2DModel"
+  ]
+}

epochs/ep205/model_index.json

@@ -0,0 +1,15 @@
+{
+  "_class_name": "DDPMCorrectorPipeline",
+  "_diffusers_version": "0.34.0",
+  "allowed_ctrl_timesteps": [
+    150
+  ],
+  "scheduler": [
+    "diffusers",
+    "FlowMatchEulerDiscreteScheduler"
+  ],
+  "unet": [
+    "diffusers",
+    "UNet2DModel"
+  ]
+}

epochs/ep220/model_index.json

@@ -0,0 +1,15 @@
+{
+  "_class_name": "DDPMCorrectorPipeline",
+  "_diffusers_version": "0.34.0",
+  "allowed_ctrl_timesteps": [
+    150
+  ],
+  "scheduler": [
+    "diffusers",
+    "FlowMatchEulerDiscreteScheduler"
+  ],
+  "unet": [
+    "diffusers",
+    "UNet2DModel"
+  ]
+}

epochs/ep230/model_index.json

@@ -0,0 +1,15 @@
+{
+  "_class_name": "DDPMCorrectorPipeline",
+  "_diffusers_version": "0.34.0",
+  "allowed_ctrl_timesteps": [
+    150
+  ],
+  "scheduler": [
+    "diffusers",
+    "FlowMatchEulerDiscreteScheduler"
+  ],
+  "unet": [
+    "diffusers",
+    "UNet2DModel"
+  ]
+}

epochs/ep235/model_index.json

@@ -0,0 +1,15 @@
+{
+  "_class_name": "DDPMCorrectorPipeline",
+  "_diffusers_version": "0.34.0",
+  "allowed_ctrl_timesteps": [
+    150
+  ],
+  "scheduler": [
+    "diffusers",
+    "FlowMatchEulerDiscreteScheduler"
+  ],
+  "unet": [
+    "diffusers",
+    "UNet2DModel"
+  ]
+}

epochs/ep245/model_index.json

@@ -0,0 +1,15 @@
+{
+  "_class_name": "DDPMCorrectorPipeline",
+  "_diffusers_version": "0.34.0",
+  "allowed_ctrl_timesteps": [
+    150
+  ],
+  "scheduler": [
+    "diffusers",
+    "FlowMatchEulerDiscreteScheduler"
+  ],
+  "unet": [
+    "diffusers",
+    "UNet2DModel"
+  ]
+}

epochs/ep25/model_index.json

@@ -0,0 +1,15 @@
+{
+  "_class_name": "DDPMCorrectorPipeline",
+  "_diffusers_version": "0.34.0",
+  "allowed_ctrl_timesteps": [
+    150
+  ],
+  "scheduler": [
+    "diffusers",
+    "FlowMatchEulerDiscreteScheduler"
+  ],
+  "unet": [
+    "diffusers",
+    "UNet2DModel"
+  ]
+}

epochs/ep260/model_index.json

@@ -0,0 +1,15 @@
+{
+  "_class_name": "DDPMCorrectorPipeline",
+  "_diffusers_version": "0.34.0",
+  "allowed_ctrl_timesteps": [
+    150
+  ],
+  "scheduler": [
+    "diffusers",
+    "FlowMatchEulerDiscreteScheduler"
+  ],
+  "unet": [
+    "diffusers",
+    "UNet2DModel"
+  ]
+}

epochs/ep280/model_index.json

@@ -0,0 +1,15 @@
+{
+  "_class_name": "DDPMCorrectorPipeline",
+  "_diffusers_version": "0.34.0",
+  "allowed_ctrl_timesteps": [
+    150
+  ],
+  "scheduler": [
+    "diffusers",
+    "FlowMatchEulerDiscreteScheduler"
+  ],
+  "unet": [
+    "diffusers",
+    "UNet2DModel"
+  ]
+}

epochs/ep290/model_index.json

@@ -0,0 +1,15 @@
+{
+  "_class_name": "DDPMCorrectorPipeline",
+  "_diffusers_version": "0.34.0",
+  "allowed_ctrl_timesteps": [
+    150
+  ],
+  "scheduler": [
+    "diffusers",
+    "FlowMatchEulerDiscreteScheduler"
+  ],
+  "unet": [
+    "diffusers",
+    "UNet2DModel"
+  ]
+}

epochs/ep295/model_index.json

@@ -0,0 +1,15 @@
+{
+  "_class_name": "DDPMCorrectorPipeline",
+  "_diffusers_version": "0.34.0",
+  "allowed_ctrl_timesteps": [
+    150
+  ],
+  "scheduler": [
+    "diffusers",
+    "FlowMatchEulerDiscreteScheduler"
+  ],
+  "unet": [
+    "diffusers",
+    "UNet2DModel"
+  ]
+}

epochs/ep300/model_index.json

@@ -0,0 +1,15 @@
+{
+  "_class_name": "DDPMCorrectorPipeline",
+  "_diffusers_version": "0.34.0",
+  "allowed_ctrl_timesteps": [
+    150
+  ],
+  "scheduler": [
+    "diffusers",
+    "FlowMatchEulerDiscreteScheduler"
+  ],
+  "unet": [
+    "diffusers",
+    "UNet2DModel"
+  ]
+}

epochs/ep310/model_index.json

@@ -0,0 +1,15 @@
+{
+  "_class_name": "DDPMCorrectorPipeline",
+  "_diffusers_version": "0.34.0",
+  "allowed_ctrl_timesteps": [
+    150
+  ],
+  "scheduler": [
+    "diffusers",
+    "FlowMatchEulerDiscreteScheduler"
+  ],
+  "unet": [
+    "diffusers",
+    "UNet2DModel"
+  ]
+}

epochs/ep320/model_index.json

@@ -0,0 +1,15 @@
+{
+  "_class_name": "DDPMCorrectorPipeline",
+  "_diffusers_version": "0.34.0",
+  "allowed_ctrl_timesteps": [
+    150
+  ],
+  "scheduler": [
+    "diffusers",
+    "FlowMatchEulerDiscreteScheduler"
+  ],
+  "unet": [
+    "diffusers",
+    "UNet2DModel"
+  ]
+}

epochs/ep335/model_index.json

@@ -0,0 +1,15 @@
+{
+  "_class_name": "DDPMCorrectorPipeline",
+  "_diffusers_version": "0.34.0",
+  "allowed_ctrl_timesteps": [
+    150
+  ],
+  "scheduler": [
+    "diffusers",
+    "FlowMatchEulerDiscreteScheduler"
+  ],
+  "unet": [
+    "diffusers",
+    "UNet2DModel"
+  ]
+}

epochs/ep350/model_index.json

@@ -0,0 +1,15 @@
+{
+  "_class_name": "DDPMCorrectorPipeline",
+  "_diffusers_version": "0.34.0",
+  "allowed_ctrl_timesteps": [
+    150
+  ],
+  "scheduler": [
+    "diffusers",
+    "FlowMatchEulerDiscreteScheduler"
+  ],
+  "unet": [
+    "diffusers",
+    "UNet2DModel"
+  ]
+}

epochs/ep355/model_index.json

@@ -0,0 +1,15 @@
+{
+  "_class_name": "DDPMCorrectorPipeline",
+  "_diffusers_version": "0.34.0",
+  "allowed_ctrl_timesteps": [
+    150
+  ],
+  "scheduler": [
+    "diffusers",
+    "FlowMatchEulerDiscreteScheduler"
+  ],
+  "unet": [
+    "diffusers",
+    "UNet2DModel"
+  ]
+}

epochs/ep360/model_index.json

@@ -0,0 +1,15 @@
+{
+  "_class_name": "DDPMCorrectorPipeline",
+  "_diffusers_version": "0.34.0",
+  "allowed_ctrl_timesteps": [
+    150
+  ],
+  "scheduler": [
+    "diffusers",
+    "FlowMatchEulerDiscreteScheduler"
+  ],
+  "unet": [
+    "diffusers",
+    "UNet2DModel"
+  ]
+}

epochs/ep370/model_index.json

@@ -0,0 +1,15 @@
+{
+  "_class_name": "DDPMCorrectorPipeline",
+  "_diffusers_version": "0.34.0",
+  "allowed_ctrl_timesteps": [
+    150
+  ],
+  "scheduler": [
+    "diffusers",
+    "FlowMatchEulerDiscreteScheduler"
+  ],
+  "unet": [
+    "diffusers",
+    "UNet2DModel"
+  ]
+}

epochs/ep375/model_index.json

@@ -0,0 +1,15 @@
+{
+  "_class_name": "DDPMCorrectorPipeline",
+  "_diffusers_version": "0.34.0",
+  "allowed_ctrl_timesteps": [
+    150
+  ],
+  "scheduler": [
+    "diffusers",
+    "FlowMatchEulerDiscreteScheduler"
+  ],
+  "unet": [
+    "diffusers",
+    "UNet2DModel"
+  ]
+}

epochs/ep385/model_index.json

@@ -0,0 +1,15 @@
+{
+  "_class_name": "DDPMCorrectorPipeline",
+  "_diffusers_version": "0.34.0",
+  "allowed_ctrl_timesteps": [
+    150
+  ],
+  "scheduler": [
+    "diffusers",
+    "FlowMatchEulerDiscreteScheduler"
+  ],
+  "unet": [
+    "diffusers",
+    "UNet2DModel"
+  ]
+}

epochs/ep405/model_index.json

@@ -0,0 +1,15 @@
+{
+  "_class_name": "DDPMCorrectorPipeline",
+  "_diffusers_version": "0.34.0",
+  "allowed_ctrl_timesteps": [
+    150
+  ],
+  "scheduler": [
+    "diffusers",
+    "FlowMatchEulerDiscreteScheduler"
+  ],
+  "unet": [
+    "diffusers",
+    "UNet2DModel"
+  ]
+}

epochs/ep425/model_index.json

@@ -0,0 +1,15 @@
+{
+  "_class_name": "DDPMCorrectorPipeline",
+  "_diffusers_version": "0.34.0",
+  "allowed_ctrl_timesteps": [
+    150
+  ],
+  "scheduler": [
+    "diffusers",
+    "FlowMatchEulerDiscreteScheduler"
+  ],
+  "unet": [
+    "diffusers",
+    "UNet2DModel"
+  ]
+}

epochs/ep430/model_index.json

@@ -0,0 +1,15 @@
+{
+  "_class_name": "DDPMCorrectorPipeline",
+  "_diffusers_version": "0.34.0",
+  "allowed_ctrl_timesteps": [
+    150
+  ],
+  "scheduler": [
+    "diffusers",
+    "FlowMatchEulerDiscreteScheduler"
+  ],
+  "unet": [
+    "diffusers",
+    "UNet2DModel"
+  ]
+}

epochs/ep440/model_index.json

@@ -0,0 +1,15 @@
+{
+  "_class_name": "DDPMCorrectorPipeline",
+  "_diffusers_version": "0.34.0",
+  "allowed_ctrl_timesteps": [
+    150
+  ],
+  "scheduler": [
+    "diffusers",
+    "FlowMatchEulerDiscreteScheduler"
+  ],
+  "unet": [
+    "diffusers",
+    "UNet2DModel"
+  ]
+}

epochs/ep470/model_index.json

@@ -0,0 +1,15 @@
+{
+  "_class_name": "DDPMCorrectorPipeline",
+  "_diffusers_version": "0.34.0",
+  "allowed_ctrl_timesteps": [
+    150
+  ],
+  "scheduler": [
+    "diffusers",
+    "FlowMatchEulerDiscreteScheduler"
+  ],
+  "unet": [
+    "diffusers",
+    "UNet2DModel"
+  ]
+}

epochs/ep475/model_index.json

@@ -0,0 +1,15 @@
+{
+  "_class_name": "DDPMCorrectorPipeline",
+  "_diffusers_version": "0.34.0",
+  "allowed_ctrl_timesteps": [
+    150
+  ],
+  "scheduler": [
+    "diffusers",
+    "FlowMatchEulerDiscreteScheduler"
+  ],
+  "unet": [
+    "diffusers",
+    "UNet2DModel"
+  ]
+}

epochs/ep485/model_index.json

@@ -0,0 +1,15 @@
+{
+  "_class_name": "DDPMCorrectorPipeline",
+  "_diffusers_version": "0.34.0",
+  "allowed_ctrl_timesteps": [
+    150
+  ],
+  "scheduler": [
+    "diffusers",
+    "FlowMatchEulerDiscreteScheduler"
+  ],
+  "unet": [
+    "diffusers",
+    "UNet2DModel"
+  ]
+}

epochs/ep495/model_index.json

@@ -0,0 +1,15 @@
+{
+  "_class_name": "DDPMCorrectorPipeline",
+  "_diffusers_version": "0.34.0",
+  "allowed_ctrl_timesteps": [
+    150
+  ],
+  "scheduler": [
+    "diffusers",
+    "FlowMatchEulerDiscreteScheduler"
+  ],
+  "unet": [
+    "diffusers",
+    "UNet2DModel"
+  ]
+}

epochs/ep500/model_index.json

@@ -0,0 +1,15 @@
+{
+  "_class_name": "DDPMCorrectorPipeline",
+  "_diffusers_version": "0.34.0",
+  "allowed_ctrl_timesteps": [
+    150
+  ],
+  "scheduler": [
+    "diffusers",
+    "FlowMatchEulerDiscreteScheduler"
+  ],
+  "unet": [
+    "diffusers",
+    "UNet2DModel"
+  ]
+}

epochs/ep55/model_index.json

@@ -0,0 +1,15 @@
+{
+  "_class_name": "DDPMCorrectorPipeline",
+  "_diffusers_version": "0.34.0",
+  "allowed_ctrl_timesteps": [
+    150
+  ],
+  "scheduler": [
+    "diffusers",
+    "FlowMatchEulerDiscreteScheduler"
+  ],
+  "unet": [
+    "diffusers",
+    "UNet2DModel"
+  ]
+}

epochs/ep60/model_index.json

@@ -0,0 +1,15 @@
+{
+  "_class_name": "DDPMCorrectorPipeline",
+  "_diffusers_version": "0.34.0",
+  "allowed_ctrl_timesteps": [
+    150
+  ],
+  "scheduler": [
+    "diffusers",
+    "FlowMatchEulerDiscreteScheduler"
+  ],
+  "unet": [
+    "diffusers",
+    "UNet2DModel"
+  ]
+}

epochs/ep70/model_index.json

@@ -0,0 +1,15 @@
+{
+  "_class_name": "DDPMCorrectorPipeline",
+  "_diffusers_version": "0.34.0",
+  "allowed_ctrl_timesteps": [
+    150
+  ],
+  "scheduler": [
+    "diffusers",
+    "FlowMatchEulerDiscreteScheduler"
+  ],
+  "unet": [
+    "diffusers",
+    "UNet2DModel"
+  ]
+}

epochs/ep90/model_index.json

@@ -0,0 +1,15 @@
+{
+  "_class_name": "DDPMCorrectorPipeline",
+  "_diffusers_version": "0.34.0",
+  "allowed_ctrl_timesteps": [
+    150
+  ],
+  "scheduler": [
+    "diffusers",
+    "FlowMatchEulerDiscreteScheduler"
+  ],
+  "unet": [
+    "diffusers",
+    "UNet2DModel"
+  ]
+}

finished.json

@@ -0,0 +1,3 @@
+{
+    "Msg": "Training finished."
+}

model_index.json

@@ -0,0 +1,15 @@
+{
+  "_class_name": "DDPMCorrectorPipeline",
+  "_diffusers_version": "0.34.0",
+  "allowed_ctrl_timesteps": [
+    150
+  ],
+  "scheduler": [
+    "diffusers",
+    "FlowMatchEulerDiscreteScheduler"
+  ],
+  "unet": [
+    "diffusers",
+    "UNet2DModel"
+  ]
+}

scheduler/scheduler_config.json

@@ -0,0 +1,18 @@
+{
+  "_class_name": "FlowMatchEulerDiscreteScheduler",
+  "_diffusers_version": "0.34.0",
+  "base_image_seq_len": 256,
+  "base_shift": 0.5,
+  "invert_sigmas": false,
+  "max_image_seq_len": 4096,
+  "max_shift": 1.15,
+  "num_train_timesteps": 1000,
+  "shift": 1.0,
+  "shift_terminal": null,
+  "stochastic_sampling": false,
+  "time_shift_type": "exponential",
+  "use_beta_sigmas": false,
+  "use_dynamic_shifting": false,
+  "use_exponential_sigmas": false,
+  "use_karras_sigmas": false
+}

training_result.json

@@ -0,0 +1,4 @@
+{
+    "best_epoch": 340,
+    "best_step": 66300
+}

unet/config.json

@@ -0,0 +1,48 @@
+{
+  "_class_name": "UNet2DModel",
+  "_diffusers_version": "0.34.0",
+  "_name_or_path": "/home/sc3379_cornell_edu/.cache/huggingface/hub/models--google--ddpm-cifar10-32/snapshots/267b167dc01f0e4e61923ea244e8b988f84deb80",
+  "act_fn": "silu",
+  "add_attention": true,
+  "attention_head_dim": null,
+  "attn_norm_num_groups": null,
+  "block_out_channels": [
+    128,
+    256,
+    256,
+    256
+  ],
+  "center_input_sample": false,
+  "class_embed_type": null,
+  "down_block_types": [
+    "DownBlock2D",
+    "AttnDownBlock2D",
+    "DownBlock2D",
+    "DownBlock2D"
+  ],
+  "downsample_padding": 0,
+  "downsample_type": "conv",
+  "dropout": 0.0,
+  "flip_sin_to_cos": false,
+  "freq_shift": 1,
+  "in_channels": 3,
+  "layers_per_block": 2,
+  "mid_block_scale_factor": 1,
+  "mid_block_type": "UNetMidBlock2D",
+  "norm_eps": 1e-06,
+  "norm_num_groups": 32,
+  "num_class_embeds": null,
+  "num_train_timesteps": null,
+  "out_channels": 3,
+  "resnet_time_scale_shift": "default",
+  "sample_size": 32,
+  "time_embedding_dim": null,
+  "time_embedding_type": "positional",
+  "up_block_types": [
+    "UpBlock2D",
+    "UpBlock2D",
+    "AttnUpBlock2D",
+    "UpBlock2D"
+  ],
+  "upsample_type": "conv"
+}