From 7917820e59446ede326d2cfc290bd5341f0424c9 Mon Sep 17 00:00:00 2001
From: Starbeamrainbowlabs <sbrl@starbeamrainbowlabs.com>
Date: Fri, 2 Sep 2022 19:11:54 +0100
Subject: [PATCH] ai: slurm fixup

---
 aimodel/slurm-pretrain.job | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/aimodel/slurm-pretrain.job b/aimodel/slurm-pretrain.job
index 9e3d404..e106a7d 100755
--- a/aimodel/slurm-pretrain.job
+++ b/aimodel/slurm-pretrain.job
@@ -62,5 +62,5 @@ echo ">>> Installing requirements";
 conda run -n py38 pip install -r requirements.txt;
 echo ">>> Training model";
 #shellcheck disable=SC2086
-/usr/bin/env time -v conda run -n py38 src/index.py --input "${CONFIG}" --output "${dir_output}" ${ARGS};
+/usr/bin/env time -v conda run -n py38 src/index.py pretrain --input "${CONFIG}" --output "${dir_output}" ${ARGS};
 echo ">>> exited with code $?";