Spaces:
Paused
Paused
nroggendorff
commited on
Commit
•
b0d3d22
1
Parent(s):
fa92db0
Update train.py
Browse files
train.py
CHANGED
@@ -16,7 +16,7 @@ VOCAB_SIZE = 32000
|
|
16 |
INPUT_DATASET = "HuggingFaceTB/smollm-corpus"
|
17 |
INSTRUCT_DATASET = "nroggendorff/elephant"
|
18 |
OUTPUT_REPO = "nroggendorff/smallama"
|
19 |
-
INSTRUCT_FINETUNE_BOOL =
|
20 |
FP16 = False
|
21 |
WARMUP_STEPS = 0
|
22 |
DECAY = 0
|
|
|
16 |
INPUT_DATASET = "HuggingFaceTB/smollm-corpus"
|
17 |
INSTRUCT_DATASET = "nroggendorff/elephant"
|
18 |
OUTPUT_REPO = "nroggendorff/smallama"
|
19 |
+
INSTRUCT_FINETUNE_BOOL = True
|
20 |
FP16 = False
|
21 |
WARMUP_STEPS = 0
|
22 |
DECAY = 0
|