From 66c29dd44f4417d77148b5087e825fbda6ad0929 Mon Sep 17 00:00:00 2001
From: Jake Poznanski <jakep@allenai.org>
Date: Tue, 24 Sep 2024 11:24:14 -0700
Subject: [PATCH] Moving to making a new dockerfile

---
 scripts/beaker/Dockerfile    | 3 +++
 scripts/qwen2vl-2b-gantry.sh | 2 +-
 2 files changed, 4 insertions(+), 1 deletion(-)
 create mode 100644 scripts/beaker/Dockerfile

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 \