| { | |
| "beta_schedule": "shifted_cosine:0.5", | |
| "clip_sample": false, | |
| "codebook_size": 8192, | |
| "conditioning": "concat", | |
| "dec_type": "uvit_b_p4_f16", | |
| "enc_type": "vit_b_enc", | |
| "image_size": 512, | |
| "image_size_dec": null, | |
| "image_size_enc": 256, | |
| "latent_dim": 32, | |
| "n_channels": 1, | |
| "n_labels": null, | |
| "norm_codes": true, | |
| "norm_latents": false, | |
| "num_codebooks": 1, | |
| "num_train_timesteps": 1000, | |
| "patch_proj": true, | |
| "patch_size": 16, | |
| "post_mlp": true, | |
| "prediction_type": "sample", | |
| "quant_type": "lucid", | |
| "scheduler": "ddim", | |
| "sync_codebook": false, | |
| "thresholding": true, | |
| "undo_std": false, | |
| "zero_terminal_snr": true | |
| } |