From c94f5d042e59a8607854552dc03700ea3c5504cc Mon Sep 17 00:00:00 2001
From: Starbeamrainbowlabs <sbrl@starbeamrainbowlabs.com>
Date: Fri, 2 Sep 2022 19:13:56 +0100
Subject: [PATCH] ai: slurm fixup again

---
 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 e106a7d..e1baa99 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 pretrain --input "${CONFIG}" --output "${dir_output}" ${ARGS};
+/usr/bin/env time -v conda run -n py38 src/index.py pretrain --input "${INPUT}" --output "${dir_output}" ${ARGS};
 echo ">>> exited with code $?";