From 15e00642126a95c735b10aefd2003a5f249dc045 Mon Sep 17 00:00:00 2001 From: Jake Poznanski Date: Thu, 29 May 2025 17:20:32 +0000 Subject: [PATCH] More fixes --- scripts/run_benchmark.sh | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/scripts/run_benchmark.sh b/scripts/run_benchmark.sh index 58cdcb0..6127c20 100755 --- a/scripts/run_benchmark.sh +++ b/scripts/run_benchmark.sh @@ -45,7 +45,7 @@ image_tag = sys.argv[1] beaker_user = sys.argv[2] # Initialize Beaker client -b = Beaker.from_env(default_workspace="ai2/oe-data-pdf") +b = Beaker.from_env(default_workspace="ai2/olmocr") # Create experiment spec experiment_spec = ExperimentSpec( @@ -69,14 +69,14 @@ experiment_spec = ExperimentSpec( preemptible=True, ), resources=TaskResources(gpu_count=1), - constraints=Constraint(cluster=["ai2/ceres-cirrascale", "ai2/jupiter-cirrascale"]), + constraints=Constraints(cluster=["ai2/ceres-cirrascale", "ai2/jupiter-cirrascale"]), result=ResultSpec(path="/noop-results"), ) ], ) # Create the experiment -experiment = b.experiment.create(spec=experiment_spec, workspace="ai2/oe-data-pdf") +experiment = b.experiment.create(spec=experiment_spec, workspace="ai2/olmocr") print(f"Created experiment: {experiment.id}") print(f"View at: https://beaker.org/ex/{experiment.id}") EOF