AlterraLaniakea commited on
Commit
1c9e58d
·
verified ·
1 Parent(s): a39b4c9

Upload model_config.yaml with huggingface_hub

Browse files
Files changed (1) hide show
  1. model_config.yaml +4 -4
model_config.yaml CHANGED
@@ -2,8 +2,8 @@ stride: 3
2
  num_frames: 16
3
  vision_config:
4
  name: vjepa2_1_vit_large_384
5
- load_from: /kaggle/input/notebooks/kaggletrainserver/vjepa2-dowload/facebookresearch_vjepa2_main
6
- source: local
7
  video_mode: true
8
  use_sdpa: true
9
  use_rope: true
@@ -16,7 +16,7 @@ vision_config:
16
  freeze: true
17
  predictor:
18
  language:
19
- model: /kaggle/input/notebooks/alterralaniakea/llama-3-2-1b
20
  num_layers: 4
21
  max_query_length: 512
22
  unload: false
@@ -45,7 +45,7 @@ decoder:
45
  freeze: false
46
  runtime:
47
  trust_remote_code: true
48
- hf_token: false
49
  torch_dtype: auto
50
  tiny: false
51
  embedding_dim: 1024
 
2
  num_frames: 16
3
  vision_config:
4
  name: vjepa2_1_vit_large_384
5
+ load_from: facebookresearch/vjepa2
6
+ source: github
7
  video_mode: true
8
  use_sdpa: true
9
  use_rope: true
 
16
  freeze: true
17
  predictor:
18
  language:
19
+ model: meta-llama/Llama-3.2-1B
20
  num_layers: 4
21
  max_query_length: 512
22
  unload: false
 
45
  freeze: false
46
  runtime:
47
  trust_remote_code: true
48
+ hf_token: true
49
  torch_dtype: auto
50
  tiny: false
51
  embedding_dim: 1024