Add ViSAGe-SC (IJCV) checkpoints and fine-tuned UniCodec
Browse filesvisage_sc: ViSAGe-SC trained on YT-Ambigen (elevation-first ordering, rotation augmentation), epoch 16 -- the model reported in the IJCV paper.
visage_sc_vgg: VGGSound-pretrained initialization used to train the above.
unicodec/final.ckpt: UniCodec fine-tuned for video-to-ambisonics generation, with optimizer and callback state stripped for inference.
- unicodec/final.ckpt +3 -0
- visage_sc/config.json +58 -0
- visage_sc/pytorch_model.bin +3 -0
- visage_sc_vgg/config.json +57 -0
- visage_sc_vgg/pytorch_model.bin +3 -0
unicodec/final.ckpt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:5668b926f6a3c8e7ca73f5ca3179a2c5fa4df57d2f9bde207e5f69c7367ed01e
|
| 3 |
+
size 1351219409
|
visage_sc/config.json
ADDED
|
@@ -0,0 +1,58 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"_name_or_path": "/gallery_millet/jaeyeon.kim/ijcv_results/unicodec_checkpoint/vgg_final",
|
| 3 |
+
"activation_dropout": 0.0,
|
| 4 |
+
"activation_function": "gelu",
|
| 5 |
+
"attention_dropout": 0.0,
|
| 6 |
+
"bos_token_id": 0,
|
| 7 |
+
"cfg_dropout_prob": 0.1,
|
| 8 |
+
"cfg_guidance_scale": 2.5,
|
| 9 |
+
"cfg_spatial_guidance_scale": 2.5,
|
| 10 |
+
"classifier_dropout": 0.0,
|
| 11 |
+
"classifier_free_guidance": true,
|
| 12 |
+
"clip_frame_rate": 4,
|
| 13 |
+
"d_clip": 512,
|
| 14 |
+
"d_model": 1024,
|
| 15 |
+
"dac_downsample_rate": 320,
|
| 16 |
+
"dac_frame_rate": 75,
|
| 17 |
+
"dac_pad_token_id": 16384,
|
| 18 |
+
"dac_sample_rate": 24000,
|
| 19 |
+
"dac_vocab_size": 16384,
|
| 20 |
+
"decoder_attention_heads": 16,
|
| 21 |
+
"decoder_ffn_dim": 4096,
|
| 22 |
+
"decoder_layerdrop": 0.0,
|
| 23 |
+
"decoder_layers": 12,
|
| 24 |
+
"decoder_start_token_id": 2,
|
| 25 |
+
"dropout": 0.1,
|
| 26 |
+
"encoder_attention_heads": 16,
|
| 27 |
+
"encoder_ffn_dim": 4096,
|
| 28 |
+
"encoder_layerdrop": 0.0,
|
| 29 |
+
"encoder_layers": 6,
|
| 30 |
+
"eos_token_id": 2,
|
| 31 |
+
"forced_eos_token_id": 2,
|
| 32 |
+
"id2label": {
|
| 33 |
+
"0": "LABEL_0",
|
| 34 |
+
"1": "LABEL_1",
|
| 35 |
+
"2": "LABEL_2"
|
| 36 |
+
},
|
| 37 |
+
"init_additional_from_clip": false,
|
| 38 |
+
"init_std": 0.02,
|
| 39 |
+
"is_encoder_decoder": true,
|
| 40 |
+
"label2id": {
|
| 41 |
+
"LABEL_0": 0,
|
| 42 |
+
"LABEL_1": 1,
|
| 43 |
+
"LABEL_2": 2
|
| 44 |
+
},
|
| 45 |
+
"label_smoothing": 0.0,
|
| 46 |
+
"max_position_embeddings": 1024,
|
| 47 |
+
"max_source_length": 80,
|
| 48 |
+
"max_target_length": 400,
|
| 49 |
+
"model_type": "bart",
|
| 50 |
+
"num_hidden_layers": 6,
|
| 51 |
+
"num_rvq": 1,
|
| 52 |
+
"pad_token_id": 1,
|
| 53 |
+
"scale_embedding": false,
|
| 54 |
+
"tie_word_embeddings": false,
|
| 55 |
+
"transformers_version": "4.40.2",
|
| 56 |
+
"use_cache": true,
|
| 57 |
+
"vocab_size": 50265
|
| 58 |
+
}
|
visage_sc/pytorch_model.bin
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:2cb353bb6b910b9161241358e1e44049c0eca6b90eee2c2744a59ed18252e097
|
| 3 |
+
size 1659624050
|
visage_sc_vgg/config.json
ADDED
|
@@ -0,0 +1,57 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"activation_dropout": 0.0,
|
| 3 |
+
"activation_function": "gelu",
|
| 4 |
+
"attention_dropout": 0.0,
|
| 5 |
+
"bos_token_id": 0,
|
| 6 |
+
"cfg_dropout_prob": 0.1,
|
| 7 |
+
"cfg_guidance_scale": 2.5,
|
| 8 |
+
"cfg_spatial_guidance_scale": 2.5,
|
| 9 |
+
"classifier_dropout": 0.0,
|
| 10 |
+
"classifier_free_guidance": true,
|
| 11 |
+
"clip_frame_rate": 4,
|
| 12 |
+
"d_clip": 512,
|
| 13 |
+
"d_model": 1024,
|
| 14 |
+
"dac_downsample_rate": 320,
|
| 15 |
+
"dac_frame_rate": 75,
|
| 16 |
+
"dac_pad_token_id": 16384,
|
| 17 |
+
"dac_sample_rate": 24000,
|
| 18 |
+
"dac_vocab_size": 16384,
|
| 19 |
+
"decoder_attention_heads": 16,
|
| 20 |
+
"decoder_ffn_dim": 4096,
|
| 21 |
+
"decoder_layerdrop": 0.0,
|
| 22 |
+
"decoder_layers": 12,
|
| 23 |
+
"decoder_start_token_id": 2,
|
| 24 |
+
"dropout": 0.1,
|
| 25 |
+
"encoder_attention_heads": 16,
|
| 26 |
+
"encoder_ffn_dim": 4096,
|
| 27 |
+
"encoder_layerdrop": 0.0,
|
| 28 |
+
"encoder_layers": 6,
|
| 29 |
+
"eos_token_id": 2,
|
| 30 |
+
"forced_eos_token_id": 2,
|
| 31 |
+
"id2label": {
|
| 32 |
+
"0": "LABEL_0",
|
| 33 |
+
"1": "LABEL_1",
|
| 34 |
+
"2": "LABEL_2"
|
| 35 |
+
},
|
| 36 |
+
"init_additional_from_clip": false,
|
| 37 |
+
"init_std": 0.02,
|
| 38 |
+
"is_encoder_decoder": true,
|
| 39 |
+
"label2id": {
|
| 40 |
+
"LABEL_0": 0,
|
| 41 |
+
"LABEL_1": 1,
|
| 42 |
+
"LABEL_2": 2
|
| 43 |
+
},
|
| 44 |
+
"label_smoothing": 0.0,
|
| 45 |
+
"max_position_embeddings": 1024,
|
| 46 |
+
"max_source_length": 80,
|
| 47 |
+
"max_target_length": 400,
|
| 48 |
+
"model_type": "bart",
|
| 49 |
+
"num_hidden_layers": 6,
|
| 50 |
+
"num_rvq": 1,
|
| 51 |
+
"pad_token_id": 1,
|
| 52 |
+
"scale_embedding": false,
|
| 53 |
+
"tie_word_embeddings": false,
|
| 54 |
+
"transformers_version": "4.40.2",
|
| 55 |
+
"use_cache": true,
|
| 56 |
+
"vocab_size": 50265
|
| 57 |
+
}
|
visage_sc_vgg/pytorch_model.bin
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:1fa9c5ae803ad5eb0903919ee0db03d64be65cc913d627de37ddfd6649115123
|
| 3 |
+
size 1247362662
|