| @@ -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.") | |||
| # ---------------------------- | |||