|
{ |
|
"_class_name": "ForwardModelUNet2DConditionModel", |
|
"_diffusers_version": "0.26.3", |
|
"_name_or_path": "ckpts/unet_lgm", |
|
"beta_end": 0.012, |
|
"beta_schedule": "scaled_linear", |
|
"beta_start": 0.00085, |
|
"clip_sample": false, |
|
"clip_sample_range": 1.0, |
|
"dynamic_thresholding_ratio": 0.995, |
|
"lgm_cam_radius": 1.5, |
|
"lgm_down_attention": [ |
|
false, |
|
false, |
|
false, |
|
true, |
|
true, |
|
true |
|
], |
|
"lgm_down_channels": [ |
|
64, |
|
128, |
|
256, |
|
512, |
|
1024, |
|
1024 |
|
], |
|
"lgm_fovy": 49.1343, |
|
"lgm_input_size": 256, |
|
"lgm_mid_attention": true, |
|
"lgm_output_size": 512, |
|
"lgm_splat_size": 128, |
|
"lgm_up_attention": [ |
|
true, |
|
true, |
|
true, |
|
false, |
|
false |
|
], |
|
"lgm_up_channels": [ |
|
1024, |
|
1024, |
|
512, |
|
256, |
|
128 |
|
], |
|
"lgm_zfar": 2.5, |
|
"lgm_znear": 0.5, |
|
"num_train_timesteps": 1000, |
|
"prediction_type": "epsilon", |
|
"rescale_betas_zero_snr": false, |
|
"sample_max_value": 1.0, |
|
"set_alpha_to_one": false, |
|
"thresholding": false, |
|
"unet_adm_in_channels": null, |
|
"unet_attention_resolutions": [ |
|
4, |
|
2, |
|
1 |
|
], |
|
"unet_camera_dim": 16, |
|
"unet_channel_mult": [ |
|
1, |
|
2, |
|
4, |
|
4 |
|
], |
|
"unet_context_dim": 1024, |
|
"unet_conv_resample": true, |
|
"unet_dims": 2, |
|
"unet_dropout": 0, |
|
"unet_image_size": 32, |
|
"unet_in_channels": 4, |
|
"unet_ip_dim": 16, |
|
"unet_ip_weight": 1.0, |
|
"unet_model_channels": 320, |
|
"unet_n_embed": null, |
|
"unet_num_attention_blocks": null, |
|
"unet_num_classes": null, |
|
"unet_num_head_channels": 64, |
|
"unet_num_heads": -1, |
|
"unet_num_heads_upsample": -1, |
|
"unet_num_res_blocks": 2, |
|
"unet_out_channels": 4, |
|
"unet_resblock_updown": false, |
|
"unet_transformer_depth": 1, |
|
"unet_use_scale_shift_norm": false, |
|
"vae_act_fn": "silu", |
|
"vae_block_out_channels": [ |
|
128, |
|
256, |
|
512, |
|
512 |
|
], |
|
"vae_down_block_types": [ |
|
"DownEncoderBlock2D", |
|
"DownEncoderBlock2D", |
|
"DownEncoderBlock2D", |
|
"DownEncoderBlock2D" |
|
], |
|
"vae_force_upcast": true, |
|
"vae_in_channels": 3, |
|
"vae_latent_channels": 4, |
|
"vae_layers_per_block": 2, |
|
"vae_norm_num_groups": 32, |
|
"vae_out_channels": 3, |
|
"vae_sample_size": 512, |
|
"vae_scaling_factor": 0.18215, |
|
"vae_up_block_types": [ |
|
"UpDecoderBlock2D", |
|
"UpDecoderBlock2D", |
|
"UpDecoderBlock2D", |
|
"UpDecoderBlock2D" |
|
] |
|
} |
|
|