diff --git a/.gitlab-ci.yml b/.gitlab-ci.yml index e3b55fd0427fe95a161ba3e8fe27586eb5d082f7..4f53698e505b939819900355e2401fc0143c67c1 100644 --- a/.gitlab-ci.yml +++ b/.gitlab-ci.yml @@ -1,5 +1,5 @@ stages: - # - train + - train - deploy variables: @@ -17,17 +17,17 @@ before_script: - pip install transformers[torch] # Ensure that all required dependencies are installed - pip install accelerate -U # Ensure the accelerate package is installed and updated -#train: - #stage: train - #tags: - # - pages - #script: - # - source venv/bin/activate - # - python train_model.py - #artifacts: - # paths: - # - trained_model.joblib - # - tokenizer.joblib +train: + stage: train + tags: + - pages + script: + - source venv/bin/activate + - python train_model.py + artifacts: + paths: + - trained_model.joblib + - tokenizer.joblib deploy: stage: deploy tags: