Update training_config.yml
Browse files- training_config.yml +9 -3
training_config.yml
CHANGED
|
@@ -34,9 +34,15 @@ top_k: 225
|
|
| 34 |
dataset:
|
| 35 |
_component_: ds.EvenBatcher
|
| 36 |
buffer_size: 1000
|
| 37 |
-
|
| 38 |
-
_component_: ds.
|
| 39 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 40 |
- _component_: ds.CaptionInstructDataset
|
| 41 |
dataset_path: ds/sam_llava/output.parquet
|
| 42 |
train_on_input: false
|
|
|
|
| 34 |
dataset:
|
| 35 |
_component_: ds.EvenBatcher
|
| 36 |
buffer_size: 1000
|
| 37 |
+
datasets:
|
| 38 |
+
- _component_: ds.OmegaVideoCaptionDataset
|
| 39 |
+
length: 500000
|
| 40 |
+
- _component_: ds.LlavaInstructDataset
|
| 41 |
+
dataset_path: ds/coco_llava_instruct/output.parquet
|
| 42 |
+
train_on_input: false
|
| 43 |
+
- _component_: ds.LlavaInstructDataset
|
| 44 |
+
dataset_path: ds/vision_flan/output.parquet
|
| 45 |
+
train_on_input: false
|
| 46 |
- _component_: ds.CaptionInstructDataset
|
| 47 |
dataset_path: ds/sam_llava/output.parquet
|
| 48 |
train_on_input: false
|