jianzongwu commited on
Commit
84de076
·
verified ·
1 Parent(s): 5c0b1a2

Upload 2 files

Browse files
unet/config.json CHANGED
@@ -1,6 +1,6 @@
1
  {
2
  "_class_name": "RoviModel",
3
- "_diffusers_version": "0.21.4",
4
  "act_fn": "silu",
5
  "attention_head_dim": 8,
6
  "block_out_channels": [
@@ -23,7 +23,7 @@
23
  "dual_cross_attention": false,
24
  "flip_sin_to_cos": true,
25
  "freq_shift": 0,
26
- "in_channels": 8,
27
  "lan_cross_attn": false,
28
  "layers_per_block": 2,
29
  "mask_decoder_type": "MaskDecoder",
@@ -37,7 +37,7 @@
37
  "ref_frame_type_2": null,
38
  "resnet_time_scale_shift": "default",
39
  "sample_size": 64,
40
- "temp_attn": true,
41
  "up_block_types": [
42
  "UpBlock3D",
43
  "CrossAttnUpBlock3D",
 
1
  {
2
  "_class_name": "RoviModel",
3
+ "_diffusers_version": "0.20.0",
4
  "act_fn": "silu",
5
  "attention_head_dim": 8,
6
  "block_out_channels": [
 
23
  "dual_cross_attention": false,
24
  "flip_sin_to_cos": true,
25
  "freq_shift": 0,
26
+ "in_channels": 4,
27
  "lan_cross_attn": false,
28
  "layers_per_block": 2,
29
  "mask_decoder_type": "MaskDecoder",
 
37
  "ref_frame_type_2": null,
38
  "resnet_time_scale_shift": "default",
39
  "sample_size": 64,
40
+ "st_attn": false,
41
  "up_block_types": [
42
  "UpBlock3D",
43
  "CrossAttnUpBlock3D",
unet/diffusion_pytorch_model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:612e9d3526fbbbafddaaf1c8aa419447b730ed8934ad300dc7e721bdd76724dd
3
  size 3734578460
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:3e6dfeef8359141c48128b087819474bbb086e9e6bc38b7fb6f29b868966c120
3
  size 3734578460