Upload training_config.yml with huggingface_hub
Browse files- training_config.yml +4 -4
training_config.yml
CHANGED
@@ -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.6
|
82 |
-
top_k:
|
83 |
quantizer: null
|
|
|
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: 170
|
81 |
temperature: 0.6
|
82 |
+
top_k: 200
|
83 |
quantizer: null
|