yujunjun/olmocr
1
0
Fork 0
mirror of https://github.com/allenai/olmocr.git synced 2025-06-27 04:00:02 +00:00
Code Issues Packages Projects Releases Wiki Activity

Single cluster fix

Browse Source
This commit is contained in:
Jake Poznanski 2024-11-15 13:30:27 -08:00
parent 2e1d0b6a90
commit eefb045859
2 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
pdelfin/beakerpipeline.py
View File

@ -688,7 +688,7 @@ def submit_beaker_job(args):
EnvVar(name="AWS_CREDENTIALS_FILE", secret=f"{owner}-AWS_CREDENTIALS_FILE"),
],
resources=TaskResources(gpu_count=1),
constraints=Constraints(cluster=args.beaker_cluster),
constraints=Constraints(cluster=args.beaker_cluster if isinstance(args.beaker_cluster, list) else [args.beaker_cluster]),
result=ResultSpec(path="/noop-results"),
)
],

1
pyproject.toml
View File

@ -73,6 +73,7 @@ dev = [
inference = [
#"sglang["all"]
#"git+https://github.com/sgl-project/sglang.git@eff468dd5a3d24646560eb044276585f7a11ac3c#subdirectory=python&egg=sglang[all]"
"beaker-py",
]
train = [