md896 commited on
Commit
a1e637f
·
1 Parent(s): 9552aaf

Avoid whoami rate-limit during job submission

Browse files

Use HF_JOB_NAMESPACE default to md896 and skip whoami lookup so jobs submit reliably when whoami-v2 is rate-limited.

Files changed (1) hide show
  1. launch_job.py +2 -2
launch_job.py CHANGED
@@ -43,7 +43,7 @@ _IMAGE = os.environ.get(
43
  "HF_JOB_IMAGE",
44
  "pytorch/pytorch:2.6.0-cuda12.4-cudnn9-devel",
45
  )
46
- _NAMESPACE = os.environ.get("HF_JOB_NAMESPACE")
47
 
48
  _SECRETS = None
49
  _local_hf_token = os.environ.get("HF_TOKEN") or os.environ.get("HUGGING_FACE_HUB_TOKEN") or get_token()
@@ -78,7 +78,7 @@ _job_env = {
78
 
79
  if __name__ == "__main__":
80
  api = HfApi()
81
- ns = _NAMESPACE or api.whoami()["name"]
82
  job = api.run_job(
83
  image=_IMAGE,
84
  command=["bash", "-lc", _bash],
 
43
  "HF_JOB_IMAGE",
44
  "pytorch/pytorch:2.6.0-cuda12.4-cudnn9-devel",
45
  )
46
+ _NAMESPACE = os.environ.get("HF_JOB_NAMESPACE", "md896")
47
 
48
  _SECRETS = None
49
  _local_hf_token = os.environ.get("HF_TOKEN") or os.environ.get("HUGGING_FACE_HUB_TOKEN") or get_token()
 
78
 
79
  if __name__ == "__main__":
80
  api = HfApi()
81
+ ns = _NAMESPACE
82
  job = api.run_job(
83
  image=_IMAGE,
84
  command=["bash", "-lc", _bash],