Update training_config.yml
Browse files- training_config.yml +3 -3
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:
|
|
@@ -77,7 +77,7 @@ inference:
|
|
| 77 |
{video}
|
| 78 |
|
| 79 |
Caption the previous video.'
|
| 80 |
-
max_new_tokens:
|
| 81 |
temperature: 0.6
|
| 82 |
top_k: 200
|
| 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:
|
|
|
|
| 77 |
{video}
|
| 78 |
|
| 79 |
Caption the previous video.'
|
| 80 |
+
max_new_tokens: 300
|
| 81 |
temperature: 0.6
|
| 82 |
top_k: 200
|
| 83 |
quantizer: null
|