Raiff1982
/

Codette-Training

Model card Files Files and versions

Raiff1982 commited on 16 days ago

Commit

967090e

·

verified ·

1 Parent(s): 8fef70b

Update train_codette_lora.py

Files changed (1) hide show

train_codette_lora.py +1 -1

train_codette_lora.py CHANGED Viewed

@@ -41,7 +41,7 @@ HF_TOKEN      = os.environ.get("HF_TOKEN", "")
 BASE_MODEL    = "meta-llama/Llama-3.2-1B-Instruct"
 ADAPTER_REPO  = "Raiff1982/codette-llama-adapter"   # where adapter is pushed
 DATA_REPO     = "Raiff1982/codette-training"
-DATA_FILE     = "codette_combined_train.jsonl"
 MAX_LEN       = 512
 EPOCHS        = 3
 BATCH         = 1

 BASE_MODEL    = "meta-llama/Llama-3.2-1B-Instruct"
 ADAPTER_REPO  = "Raiff1982/codette-llama-adapter"   # where adapter is pushed
 DATA_REPO     = "Raiff1982/codette-training"
+DATA_FILE     = "codette_music_training.jsonl"
 MAX_LEN       = 512
 EPOCHS        = 3
 BATCH         = 1