Upload training_config.yml with huggingface_hub
Browse files- training_config.yml +6 -0
training_config.yml
CHANGED
|
@@ -62,3 +62,9 @@ enable_activation_checkpointing: false
|
|
| 62 |
profiler:
|
| 63 |
_component_: torchtune.utils.profiler
|
| 64 |
enabled: false
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 62 |
profiler:
|
| 63 |
_component_: torchtune.utils.profiler
|
| 64 |
enabled: false
|
| 65 |
+
inference:
|
| 66 |
+
prompt_template: "Video:\n{video}\nCaption the previous video."
|
| 67 |
+
max_new_tokens: 300
|
| 68 |
+
temperature: 0.6 # 0.8 and 0.6 are popular values to try
|
| 69 |
+
top_k: 300
|
| 70 |
+
quantizer: null
|