diff --git a/Finetunning/finetunning.py b/Finetunning/finetunning.py index 4b9e80f..cf8f687 100644 --- a/Finetunning/finetunning.py +++ b/Finetunning/finetunning.py @@ -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...") -trainer.train(resume_from_checkpoint=True) +try: + trainer.train(resume_from_checkpoint=True) +except: + trainer.train() print("Training completed successfully.") # ----------------------------