Spaces:

nroggendorff
/

train-llama

Paused

nroggendorff commited on Sep 2, 2024

Commit

8cbd82e

verified ·

1 Parent(s): 4c1f178

Update train.py

Files changed (1) hide show

train.py CHANGED Viewed

@@ -15,7 +15,7 @@ MAX_SEQ_LENGTH = 128
 VOCAB_SIZE = 32000
 INPUT_DATASET = "HuggingFaceTB/smollm-corpus"
 INSTRUCT_DATASET = "nroggendorff/elephant"
-OUTPUT_REPO = "smallama"
 INSTRUCT_FINETUNE_BOOL = True
 FP16 = True
 WARMUP_STEPS = 0

 VOCAB_SIZE = 32000
 INPUT_DATASET = "HuggingFaceTB/smollm-corpus"
 INSTRUCT_DATASET = "nroggendorff/elephant"
+OUTPUT_REPO = "nroggendorff/smallama"
 INSTRUCT_FINETUNE_BOOL = True
 FP16 = True
 WARMUP_STEPS = 0