Commit
·
f6fddf9
1
Parent(s):
f5b2bd6
fix
Browse files
configs/Qwen3-4B-Base_a10g-small.yaml
CHANGED
|
@@ -5,7 +5,7 @@ torch_dtype: bfloat16
|
|
| 5 |
attn_implementation: flash_attention_2
|
| 6 |
|
| 7 |
# Data training arguments
|
| 8 |
-
dataset_name: trl-lib/
|
| 9 |
dataset_num_proc: 4
|
| 10 |
|
| 11 |
# SFT trainer config
|
|
|
|
| 5 |
attn_implementation: flash_attention_2
|
| 6 |
|
| 7 |
# Data training arguments
|
| 8 |
+
dataset_name: trl-lib/tldr
|
| 9 |
dataset_num_proc: 4
|
| 10 |
|
| 11 |
# SFT trainer config
|