Added restart script and fixed dockerfile and run to match docs

2025-03-01 13:29:17 +01:00
parent 1d5b687e00
commit b7e95d7398
30 changed files with 3047 additions and 1640 deletions
--- a/tools/train/train_mistral.py
+++ b/tools/train/train_mistral.py
@@ -23,7 +23,7 @@ class Config:
        parser.add_argument(
            '--config',
            type=Path,
-            default=os.getenv('SIA_TRAINING_CONFIG', 'training/config.yaml'),
+            default=os.getenv('SIA_TRAINING_CONFIG', '/root/sia/training/config.yaml'),
            help='Path to config file'
        )
        parser.add_argument(