diff --git a/scripts/beaker/Dockerfile b/scripts/beaker/Dockerfile new file mode 100644 index 0000000..9633f09 --- /dev/null +++ b/scripts/beaker/Dockerfile @@ -0,0 +1,3 @@ +FROM gcr.io/ai2-beaker-core/public/cqgl31u2ba5vrtuc91og:latest + +RUN apt update && apt-get install -y libaio-dev \ No newline at end of file diff --git a/scripts/qwen2vl-2b-gantry.sh b/scripts/qwen2vl-2b-gantry.sh index aa60f27..623c9cf 100755 --- a/scripts/qwen2vl-2b-gantry.sh +++ b/scripts/qwen2vl-2b-gantry.sh @@ -27,7 +27,7 @@ gantry run \ --allow-dirty \ --host-networking \ --workspace ai2/oe-data-pdf \ - --beaker-image 'lucas/refine-axelot-vllm' \ + --beaker-image 'jakep/jakep-pdf-finetune' \ --venv 'base' \ --pip gantry-requirements.txt \ --priority normal \