krahets commited on
Commit
cb6c694
·
verified ·
1 Parent(s): 2b5bb7d

Upload folder using huggingface_hub

Browse files
model_index.json ADDED
@@ -0,0 +1,16 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_class_name": "Diffuman4DPipeline",
3
+ "_diffusers_version": "0.33.1",
4
+ "scheduler": [
5
+ "diffusers",
6
+ "DPMSolverMultistepScheduler"
7
+ ],
8
+ "unet": [
9
+ "src.diffusers.models.unets.unet_multiview_condition",
10
+ "UNetMultiviewConditionModel"
11
+ ],
12
+ "vae": [
13
+ "diffusers",
14
+ "AutoencoderKL"
15
+ ]
16
+ }
scheduler/scheduler_config.json ADDED
@@ -0,0 +1,33 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_class_name": "DPMSolverMultistepScheduler",
3
+ "_diffusers_version": "0.33.1",
4
+ "algorithm_type": "dpmsolver++",
5
+ "beta_end": 0.012,
6
+ "beta_schedule": "scaled_linear",
7
+ "beta_start": 0.00085,
8
+ "clip_sample": false,
9
+ "dynamic_thresholding_ratio": 0.995,
10
+ "euler_at_final": false,
11
+ "final_sigmas_type": "zero",
12
+ "flow_shift": 1.0,
13
+ "lambda_min_clipped": -Infinity,
14
+ "lower_order_final": true,
15
+ "num_train_timesteps": 1000,
16
+ "prediction_type": "v_prediction",
17
+ "rescale_betas_zero_snr": false,
18
+ "sample_max_value": 1.0,
19
+ "set_alpha_to_one": false,
20
+ "skip_prk_steps": true,
21
+ "solver_order": 2,
22
+ "solver_type": "midpoint",
23
+ "steps_offset": 1,
24
+ "thresholding": false,
25
+ "timestep_spacing": "linspace",
26
+ "trained_betas": null,
27
+ "use_beta_sigmas": false,
28
+ "use_exponential_sigmas": false,
29
+ "use_flow_sigmas": false,
30
+ "use_karras_sigmas": false,
31
+ "use_lu_lambdas": false,
32
+ "variance_type": null
33
+ }
unet/config.json ADDED
@@ -0,0 +1,76 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_class_name": "UNetMultiviewConditionModel",
3
+ "_diffusers_version": "0.33.1",
4
+ "_name_or_path": "diffuman4d/unet",
5
+ "act_fn": "silu",
6
+ "addition_embed_type": null,
7
+ "addition_embed_type_num_heads": 64,
8
+ "addition_time_embed_dim": null,
9
+ "attention_head_dim": [
10
+ 5,
11
+ 10,
12
+ 20,
13
+ 20
14
+ ],
15
+ "attention_type": "default",
16
+ "block_out_channels": [
17
+ 320,
18
+ 640,
19
+ 1280,
20
+ 1280
21
+ ],
22
+ "center_input_sample": false,
23
+ "class_embed_type": null,
24
+ "class_embeddings_concat": false,
25
+ "conv_in_kernel": 3,
26
+ "conv_out_kernel": 3,
27
+ "cross_attention_dim": null,
28
+ "cross_attention_norm": null,
29
+ "down_block_types": [
30
+ "CrossAttnDownBlockMultiview",
31
+ "CrossAttnDownBlockMultiview",
32
+ "CrossAttnDownBlockMultiview",
33
+ "DownBlock2D"
34
+ ],
35
+ "downsample_padding": 1,
36
+ "dropout": 0.0,
37
+ "dual_cross_attention": false,
38
+ "enable_pose_encoder": true,
39
+ "enable_tem_embeds": true,
40
+ "encoder_hid_dim": null,
41
+ "encoder_hid_dim_type": null,
42
+ "flip_sin_to_cos": true,
43
+ "freq_shift": 0,
44
+ "in_channels": 11,
45
+ "layers_per_block": 2,
46
+ "mid_block_only_cross_attention": null,
47
+ "mid_block_scale_factor": 1,
48
+ "mid_block_type": "UNetMidBlockMultiviewCrossAttn",
49
+ "norm_eps": 1e-05,
50
+ "norm_num_groups": 32,
51
+ "num_3d_attn_blocks": 3,
52
+ "num_attention_heads": null,
53
+ "num_class_embeds": null,
54
+ "only_cross_attention": false,
55
+ "out_channels": 4,
56
+ "projection_class_embeddings_input_dim": null,
57
+ "resnet_out_scale_factor": 1.0,
58
+ "resnet_skip_time_act": false,
59
+ "resnet_time_scale_shift": "default",
60
+ "reverse_transformer_layers_per_block": null,
61
+ "sample_size": 64,
62
+ "time_cond_proj_dim": null,
63
+ "time_embedding_act_fn": null,
64
+ "time_embedding_dim": null,
65
+ "time_embedding_type": "positional",
66
+ "timestep_post_act": null,
67
+ "transformer_layers_per_block": 1,
68
+ "up_block_types": [
69
+ "UpBlock2D",
70
+ "CrossAttnUpBlockMultiview",
71
+ "CrossAttnUpBlockMultiview",
72
+ "CrossAttnUpBlockMultiview"
73
+ ],
74
+ "upcast_attention": false,
75
+ "use_linear_projection": true
76
+ }
unet/diffusion_pytorch_model.fp16.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:7804f50f0dd092c32d1f8f9431fe9030fdb93c87cdc8565559aa742fc496a43b
3
+ size 1635689818
unet/diffusion_pytorch_model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:541e5e96c265473e76f58af0dd18ee0473f6be55381854a6f43d4d43082d4655
3
+ size 3271309572
vae/config.json ADDED
@@ -0,0 +1,38 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_class_name": "AutoencoderKL",
3
+ "_diffusers_version": "0.33.1",
4
+ "_name_or_path": "stable-diffusion-2-1-base/vae",
5
+ "act_fn": "silu",
6
+ "block_out_channels": [
7
+ 128,
8
+ 256,
9
+ 512,
10
+ 512
11
+ ],
12
+ "down_block_types": [
13
+ "DownEncoderBlock2D",
14
+ "DownEncoderBlock2D",
15
+ "DownEncoderBlock2D",
16
+ "DownEncoderBlock2D"
17
+ ],
18
+ "force_upcast": true,
19
+ "in_channels": 3,
20
+ "latent_channels": 4,
21
+ "latents_mean": null,
22
+ "latents_std": null,
23
+ "layers_per_block": 2,
24
+ "mid_block_add_attention": true,
25
+ "norm_num_groups": 32,
26
+ "out_channels": 3,
27
+ "sample_size": 768,
28
+ "scaling_factor": 0.18215,
29
+ "shift_factor": null,
30
+ "up_block_types": [
31
+ "UpDecoderBlock2D",
32
+ "UpDecoderBlock2D",
33
+ "UpDecoderBlock2D",
34
+ "UpDecoderBlock2D"
35
+ ],
36
+ "use_post_quant_conv": true,
37
+ "use_quant_conv": true
38
+ }
vae/diffusion_pytorch_model.fp16.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:3e4c08995484ee61270175e9e7a072b66a6e4eeb5f0c266667fe1f45b90daf9a
3
+ size 167335342
vae/diffusion_pytorch_model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:2aa1f43011b553a4cba7f37456465cdbd48aab7b54b9348b890e8058ea7683ec
3
+ size 334643268