|
|
|
@ -88,7 +88,7 @@ print("LoRA adapters attached.") |
|
|
|
print(f"{80 * '_'}\n[5/7] Loading dataset from JSON file...") |
|
|
|
dataset = load_dataset( |
|
|
|
"json", |
|
|
|
data_files="traductions.json" |
|
|
|
data_files="paires.json" |
|
|
|
) |
|
|
|
|
|
|
|
print(f"Dataset loaded with {len(dataset['train'])} samples.") |
|
|
|
@ -147,7 +147,10 @@ print("Trainer initialized.") |
|
|
|
# Train |
|
|
|
# ---------------------------- |
|
|
|
print(f"{80 * '_'}\n[7/7] Starting training...") |
|
|
|
try: |
|
|
|
trainer.train(resume_from_checkpoint=True) |
|
|
|
except: |
|
|
|
trainer.train() |
|
|
|
print("Training completed successfully.") |
|
|
|
|
|
|
|
# ---------------------------- |
|
|
|
|