Yehonathan Litman commited on
Commit
eff733b
·
1 Parent(s): 8a9c30d

initial commit

Browse files
model_index.json ADDED
@@ -0,0 +1,21 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_class_name": "RelightingPipelineMVVAE",
3
+ "_diffusers_version": "0.32.1",
4
+ "requires_safety_checker": false,
5
+ "safety_checker": [
6
+ null,
7
+ null
8
+ ],
9
+ "scheduler": [
10
+ "diffusers",
11
+ "DDPMScheduler"
12
+ ],
13
+ "unet": [
14
+ "models.unet_2d_condition",
15
+ "UNet2DConditionModel"
16
+ ],
17
+ "vae": [
18
+ "diffusers",
19
+ "AutoencoderKL"
20
+ ]
21
+ }
scheduler/scheduler_config.json ADDED
@@ -0,0 +1,21 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_class_name": "DDPMScheduler",
3
+ "_diffusers_version": "0.32.1",
4
+ "beta_end": 0.012,
5
+ "beta_schedule": "scaled_linear",
6
+ "beta_start": 0.00085,
7
+ "clip_sample": false,
8
+ "clip_sample_range": 1.0,
9
+ "dynamic_thresholding_ratio": 0.995,
10
+ "num_train_timesteps": 1000,
11
+ "prediction_type": "v_prediction",
12
+ "rescale_betas_zero_snr": false,
13
+ "sample_max_value": 1.0,
14
+ "set_alpha_to_one": false,
15
+ "skip_prk_steps": true,
16
+ "steps_offset": 1,
17
+ "thresholding": false,
18
+ "timestep_spacing": "leading",
19
+ "trained_betas": null,
20
+ "variance_type": "fixed_small"
21
+ }
unet/config.json ADDED
@@ -0,0 +1,76 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_class_name": "UNet2DConditionModel",
3
+ "_diffusers_version": "0.32.1",
4
+ "act_fn": "silu",
5
+ "addition_embed_type": null,
6
+ "addition_embed_type_num_heads": 64,
7
+ "addition_time_embed_dim": null,
8
+ "additional_unet_kwargs": {
9
+ "lighting_cross_attn": true,
10
+ "multiview_self_attn": true
11
+ },
12
+ "attention_head_dim": [
13
+ 5,
14
+ 10,
15
+ 20,
16
+ 20
17
+ ],
18
+ "attention_type": "default",
19
+ "block_out_channels": [
20
+ 320,
21
+ 640,
22
+ 1280,
23
+ 1280
24
+ ],
25
+ "center_input_sample": false,
26
+ "class_embed_type": null,
27
+ "class_embeddings_concat": false,
28
+ "conv_in_kernel": 3,
29
+ "conv_out_kernel": 3,
30
+ "cross_attention_dim": 1024,
31
+ "cross_attention_norm": null,
32
+ "down_block_types": [
33
+ "CrossAttnDownBlock2D",
34
+ "CrossAttnDownBlock2D",
35
+ "CrossAttnDownBlock2D",
36
+ "DownBlock2D"
37
+ ],
38
+ "downsample_padding": 1,
39
+ "dropout": 0.0,
40
+ "dual_cross_attention": false,
41
+ "encoder_hid_dim": null,
42
+ "encoder_hid_dim_type": null,
43
+ "flip_sin_to_cos": true,
44
+ "freq_shift": 0,
45
+ "in_channels": 22,
46
+ "layers_per_block": 2,
47
+ "mid_block_only_cross_attention": null,
48
+ "mid_block_scale_factor": 1,
49
+ "mid_block_type": "UNetMidBlock2DCrossAttn",
50
+ "norm_eps": 1e-05,
51
+ "norm_num_groups": 32,
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
+ "CrossAttnUpBlock2D",
71
+ "CrossAttnUpBlock2D",
72
+ "CrossAttnUpBlock2D"
73
+ ],
74
+ "upcast_attention": false,
75
+ "use_linear_projection": true
76
+ }
unet/diffusion_pytorch_model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:58318c43bb2e6f7cb86f1dc429a561f2ebf69bf6a78f900e3b40693f2eb4d0c5
3
+ size 4459218808
vae/config.json ADDED
@@ -0,0 +1,37 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_class_name": "AutoencoderKL",
3
+ "_diffusers_version": "0.32.1",
4
+ "act_fn": "silu",
5
+ "block_out_channels": [
6
+ 128,
7
+ 256,
8
+ 512,
9
+ 512
10
+ ],
11
+ "down_block_types": [
12
+ "DownEncoderBlock2D",
13
+ "DownEncoderBlock2D",
14
+ "DownEncoderBlock2D",
15
+ "DownEncoderBlock2D"
16
+ ],
17
+ "force_upcast": true,
18
+ "in_channels": 3,
19
+ "latent_channels": 4,
20
+ "latents_mean": null,
21
+ "latents_std": null,
22
+ "layers_per_block": 2,
23
+ "mid_block_add_attention": true,
24
+ "norm_num_groups": 32,
25
+ "out_channels": 3,
26
+ "sample_size": 768,
27
+ "scaling_factor": 0.18215,
28
+ "shift_factor": null,
29
+ "up_block_types": [
30
+ "UpDecoderBlock2D",
31
+ "UpDecoderBlock2D",
32
+ "UpDecoderBlock2D",
33
+ "UpDecoderBlock2D"
34
+ ],
35
+ "use_post_quant_conv": true,
36
+ "use_quant_conv": true
37
+ }
vae/diffusion_pytorch_model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:8d0c34f57abe50f323040f2366c8e22b941068dcdf53c8eb1d6fafb838afecb7
3
+ size 167335590