unet_lgm / config.json
lehduong's picture
Upload model
228a652 verified
{
"_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"
]
}