cilantro9246
commited on
Commit
•
7edd610
1
Parent(s):
8131f9b
Upload training_config.yml with huggingface_hub
Browse files- training_config.yml +7 -7
training_config.yml
CHANGED
@@ -5,8 +5,8 @@ model:
|
|
5 |
- v_proj
|
6 |
apply_lora_to_mlp: false
|
7 |
apply_lora_to_output: false
|
8 |
-
lora_rank:
|
9 |
-
lora_alpha:
|
10 |
perception_tokens: 2
|
11 |
use_clip: false
|
12 |
tokenizer:
|
@@ -24,9 +24,9 @@ checkpointer:
|
|
24 |
resume_from_checkpoint: false
|
25 |
interim_checkpoint_steps: 5000
|
26 |
interim_gen_steps: null
|
27 |
-
max_new_tokens:
|
28 |
temperature: 0.8
|
29 |
-
top_k:
|
30 |
dataset:
|
31 |
_component_: ds.EvenBatcher
|
32 |
buffer_size: 36
|
@@ -77,7 +77,7 @@ inference:
|
|
77 |
{video}
|
78 |
|
79 |
Caption the previous video.'
|
80 |
-
max_new_tokens:
|
81 |
-
temperature: 0.
|
82 |
-
top_k:
|
83 |
quantizer: null
|
|
|
5 |
- v_proj
|
6 |
apply_lora_to_mlp: false
|
7 |
apply_lora_to_output: false
|
8 |
+
lora_rank: 32
|
9 |
+
lora_alpha: 64
|
10 |
perception_tokens: 2
|
11 |
use_clip: false
|
12 |
tokenizer:
|
|
|
24 |
resume_from_checkpoint: false
|
25 |
interim_checkpoint_steps: 5000
|
26 |
interim_gen_steps: null
|
27 |
+
max_new_tokens: 170
|
28 |
temperature: 0.8
|
29 |
+
top_k: 200
|
30 |
dataset:
|
31 |
_component_: ds.EvenBatcher
|
32 |
buffer_size: 36
|
|
|
77 |
{video}
|
78 |
|
79 |
Caption the previous video.'
|
80 |
+
max_new_tokens: 300
|
81 |
+
temperature: 0.7
|
82 |
+
top_k: 250
|
83 |
quantizer: null
|