Skip to content
Snippets Groups Projects
Commit 4ef3b2f7 authored by Hafiz, Aqib (PG/T - Comp Sci & Elec Eng)'s avatar Hafiz, Aqib (PG/T - Comp Sci & Elec Eng)
Browse files

Update .gitlab-ci.yml file

parent 81b62eab
No related branches found
No related tags found
No related merge requests found
Pipeline #72733 passed
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:
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment