Skip to content

Commit 9539eab

Browse files
committed
AWs creds fix
1 parent e0fda1a commit 9539eab

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

scripts/run_benchmark.sh

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -108,8 +108,8 @@ except:
108108
commands = []
109109
if has_aws_creds:
110110
commands.extend([
111-
"mkdir -p /home/ubuntu/.aws",
112-
'echo "$AWS_CREDENTIALS_FILE" > /home/ubuntu/.aws/credentials'
111+
"mkdir -p ~/.aws",
112+
'echo "$AWS_CREDENTIALS_FILE" > ~/.aws/credentials'
113113
])
114114
commands.extend([
115115
"git clone https://huggingface.co/datasets/allenai/olmOCR-bench",
@@ -132,7 +132,7 @@ task_spec_args = {
132132
preemptible=True,
133133
),
134134
"resources": TaskResources(gpu_count=1),
135-
"constraints": Constraints(cluster=["ai2/ceres-cirrascale", "ai2/jupiter-cirrascale-2"]),
135+
"constraints": Constraints(cluster=["ai2/ceres-cirrascale", "ai2/jupiter-cirrascale-2", "ai2/saturn-cirrascale", "ai2/neptune-cirrascale"]),
136136
"result": ResultSpec(path="/noop-results"),
137137
}
138138

0 commit comments

Comments
 (0)