From aad8428dc31f701db79e98c9a02cf5c66518d8f1 Mon Sep 17 00:00:00 2001 From: Jake Poznanski Date: Mon, 2 Jun 2025 23:05:48 +0000 Subject: [PATCH] Reverting custom pipeline image --- olmocr/pipeline.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/olmocr/pipeline.py b/olmocr/pipeline.py index 965eb4c..0899d77 100644 --- a/olmocr/pipeline.py +++ b/olmocr/pipeline.py @@ -740,8 +740,7 @@ def submit_beaker_job(args): b = Beaker.from_env(default_workspace=args.beaker_workspace) account = b.account.whoami() owner = account.name - #beaker_image = f"jakep/olmocr-inference-{VERSION}" - beaker_image = "jakep/olmocr-benchmark-0.1.71-d71703317d" + beaker_image = f"jakep/olmocr-inference-{VERSION}" task_name = f"olmocr-{os.path.basename(args.workspace.rstrip('/'))}"