hf-transformers-bot commited on
Commit
9bf11d2
·
verified ·
1 Parent(s): 6c2c3a7

Update tiny models for PeVideoModel

Browse files
config.json ADDED
@@ -0,0 +1,110 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "architectures": [
3
+ "PeVideoModel"
4
+ ],
5
+ "dtype": "float32",
6
+ "model_type": "pe_video",
7
+ "projection_dim": 32,
8
+ "text_config": {
9
+ "attention_bias": false,
10
+ "attention_dropout": 0.0,
11
+ "bos_token_id": 50281,
12
+ "classifier_activation": "gelu",
13
+ "classifier_bias": false,
14
+ "classifier_dropout": 0.0,
15
+ "classifier_pooling": "cls",
16
+ "cls_token_id": 50281,
17
+ "decoder_bias": true,
18
+ "deterministic_flash_attn": false,
19
+ "embedding_dropout": 0.0,
20
+ "eos_token_id": 50282,
21
+ "hidden_activation": "gelu",
22
+ "hidden_size": 32,
23
+ "initializer_cutoff_factor": 2.0,
24
+ "initializer_range": 0.02,
25
+ "intermediate_size": 37,
26
+ "is_decoder": false,
27
+ "layer_types": [
28
+ "full_attention",
29
+ "sliding_attention"
30
+ ],
31
+ "local_attention": 128,
32
+ "max_position_embeddings": 512,
33
+ "mlp_bias": false,
34
+ "mlp_dropout": 0.0,
35
+ "model_type": "modernbert",
36
+ "norm_bias": false,
37
+ "norm_eps": 1e-05,
38
+ "num_attention_heads": 4,
39
+ "num_hidden_layers": 2,
40
+ "pad_token_id": 50283,
41
+ "rope_parameters": {
42
+ "full_attention": {
43
+ "rope_theta": 160000.0,
44
+ "rope_type": "default"
45
+ },
46
+ "sliding_attention": {
47
+ "rope_theta": 10000.0,
48
+ "rope_type": "default"
49
+ }
50
+ },
51
+ "sep_token_id": 50282,
52
+ "sparse_pred_ignore_index": -100,
53
+ "sparse_prediction": false,
54
+ "tie_word_embeddings": true,
55
+ "type_vocab_size": 16,
56
+ "vocab_size": 50368
57
+ },
58
+ "transformers_version": "5.16.0.dev0",
59
+ "video_config": {
60
+ "attention_bias": false,
61
+ "attention_dropout": 0.0,
62
+ "head_dim": 16,
63
+ "hidden_act": "silu",
64
+ "hidden_size": 32,
65
+ "initializer_range": 0.02,
66
+ "intermediate_size": 37,
67
+ "max_position_embeddings": 512,
68
+ "max_window_layers": 28,
69
+ "model_type": "pe_video_encoder",
70
+ "num_attention_heads": 2,
71
+ "num_hidden_layers": 2,
72
+ "num_key_value_heads": 2,
73
+ "rms_norm_eps": 1e-05,
74
+ "rope_parameters": {
75
+ "rope_theta": 20000,
76
+ "rope_type": "default"
77
+ },
78
+ "vision_config": {
79
+ "_name_or_path": "",
80
+ "architecture": "vit_pe_core_large_patch14_336",
81
+ "architectures": null,
82
+ "chunk_size_feed_forward": 0,
83
+ "do_pooling": true,
84
+ "dtype": null,
85
+ "global_pool": "map",
86
+ "initializer_range": 0.02,
87
+ "is_encoder_decoder": false,
88
+ "label_names": [
89
+ "LABEL_0",
90
+ "LABEL_1",
91
+ "LABEL_2",
92
+ "LABEL_3"
93
+ ],
94
+ "model_args": {
95
+ "depth": 2,
96
+ "embed_dim": 64,
97
+ "img_size": [
98
+ 14,
99
+ 14
100
+ ]
101
+ },
102
+ "model_type": "timm_wrapper",
103
+ "num_classes": 4,
104
+ "output_attentions": false,
105
+ "output_hidden_states": false,
106
+ "problem_type": null,
107
+ "return_dict": true
108
+ }
109
+ }
110
+ }
model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:7b03ab9ca980e47dc006ce12c7aab34d880642b3ea8b85065d63fe266e5b6d5d
3
+ size 7379360
preprocessor_config.json ADDED
@@ -0,0 +1,9 @@
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "feature_extractor_type": "PeAudioFeatureExtractor",
3
+ "feature_size": 1,
4
+ "hop_length": 1920,
5
+ "padding_side": "right",
6
+ "padding_value": 0.0,
7
+ "return_attention_mask": true,
8
+ "sampling_rate": 48000
9
+ }
tokenizer.json ADDED
The diff for this file is too large to render. See raw diff
 
tokenizer_config.json ADDED
@@ -0,0 +1,19 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "backend": "tokenizers",
3
+ "clean_up_tokenization_spaces": true,
4
+ "cls_token": "[CLS]",
5
+ "extra_special_tokens": [],
6
+ "is_local": true,
7
+ "local_files_only": false,
8
+ "mask_token": "[MASK]",
9
+ "model_input_names": [
10
+ "input_ids",
11
+ "attention_mask"
12
+ ],
13
+ "model_max_length": 8192,
14
+ "pad_token": "[PAD]",
15
+ "processor_class": "PeAudioVideoProcessor",
16
+ "sep_token": "[SEP]",
17
+ "tokenizer_class": "TokenizersBackend",
18
+ "unk_token": "[UNK]"
19
+ }
video_preprocessor_config.json ADDED
@@ -0,0 +1,27 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "data_format": "channels_first",
3
+ "default_to_square": true,
4
+ "do_convert_rgb": true,
5
+ "do_normalize": true,
6
+ "do_rescale": true,
7
+ "do_resize": true,
8
+ "do_sample_frames": false,
9
+ "image_mean": [
10
+ 0.5,
11
+ 0.5,
12
+ 0.5
13
+ ],
14
+ "image_std": [
15
+ 0.5,
16
+ 0.5,
17
+ 0.5
18
+ ],
19
+ "resample": 2,
20
+ "rescale_factor": 0.00392156862745098,
21
+ "return_metadata": false,
22
+ "size": {
23
+ "height": 336,
24
+ "width": 336
25
+ },
26
+ "video_processor_type": "PeVideoVideoProcessor"
27
+ }