NagaSaiAbhinay commited on
Commit
0cccc31
1 Parent(s): 50f6abe

Update config.json to match latest model

Browse files
Files changed (1) hide show
  1. unet/config.json +5 -8
unet/config.json CHANGED
@@ -1,6 +1,7 @@
1
  {
2
  "_class_name": "UNet3DConditionModel",
3
- "_diffusers_version": "0.17.0.dev0",
 
4
  "act_fn": "silu",
5
  "attention_head_dim": 8,
6
  "block_out_channels": [
@@ -9,7 +10,6 @@
9
  1280,
10
  1280
11
  ],
12
- "center_input_sample": false,
13
  "cross_attention_dim": 768,
14
  "down_block_types": [
15
  "CrossAttnDownBlockInflated3D",
@@ -21,12 +21,11 @@
21
  "in_channels": 4,
22
  "layers_per_block": 2,
23
  "mid_block_scale_factor": 1,
24
- "mid_block_type" : "UNetMidBlockInflated3DCrossAttn",
25
  "norm_eps": 1e-05,
26
  "norm_num_groups": 32,
27
- "only_cross_attention": false,
28
  "out_channels": 4,
29
- "resnet_time_scale_shift": "default",
30
  "sample_size": 64,
31
  "up_block_types": [
32
  "UpBlockInflated3D",
@@ -34,7 +33,5 @@
34
  "CrossAttnUpBlockInflated3D",
35
  "CrossAttnUpBlockInflated3D"
36
  ],
37
- "upcast_attention": false,
38
- "use_linear_projection": false,
39
  "use_temporal_transformer": false
40
- }
 
1
  {
2
  "_class_name": "UNet3DConditionModel",
3
+ "_diffusers_version": "0.22.0.dev0",
4
+ "_name_or_path": "/root/.cache/huggingface/hub/models--Tune-A-Video-library--df-cpt-mo-di-bear-guitar/snapshots/50f6abec0190a7dfa07f015aa20ea9d0b9ef96e6/unet",
5
  "act_fn": "silu",
6
  "attention_head_dim": 8,
7
  "block_out_channels": [
 
10
  1280,
11
  1280
12
  ],
 
13
  "cross_attention_dim": 768,
14
  "down_block_types": [
15
  "CrossAttnDownBlockInflated3D",
 
21
  "in_channels": 4,
22
  "layers_per_block": 2,
23
  "mid_block_scale_factor": 1,
24
+ "mid_block_type": "UNetMidBlockInflated3DCrossAttn",
25
  "norm_eps": 1e-05,
26
  "norm_num_groups": 32,
27
+ "num_attention_heads": null,
28
  "out_channels": 4,
 
29
  "sample_size": 64,
30
  "up_block_types": [
31
  "UpBlockInflated3D",
 
33
  "CrossAttnUpBlockInflated3D",
34
  "CrossAttnUpBlockInflated3D"
35
  ],
 
 
36
  "use_temporal_transformer": false
37
+ }