Spaces:

ycwhencpp
/

final-iteration

Paused

vaibhav12332112312 commited on 12 days ago

Commit

76b19bd

1 Parent(s): 9ee7a09

train: default HF Job flavor l4x1 -> l40sx1 (48GB VRAM)

Files changed (1) hide show

training/hf_run_space_train_job.sh CHANGED Viewed

@@ -8,7 +8,7 @@
 set -euo pipefail
 IMAGE="${HF_JOB_IMAGE:-pytorch/pytorch:2.5.1-cuda12.4-cudnn9-runtime}"
-FLAVOR="${HF_JOB_FLAVOR:-l4x1}"
 TIMEOUT="${HF_JOB_TIMEOUT:-8h}"
 SPACE_REPO="${HF_SPACE_REPO_ID:-vaibhavkhandare/train-bhai-train}"
 NB_EXEC_TIMEOUT="${NB_EXEC_TIMEOUT:-3600}"

 set -euo pipefail
 IMAGE="${HF_JOB_IMAGE:-pytorch/pytorch:2.5.1-cuda12.4-cudnn9-runtime}"
+FLAVOR="${HF_JOB_FLAVOR:-l40sx1}"
 TIMEOUT="${HF_JOB_TIMEOUT:-8h}"
 SPACE_REPO="${HF_SPACE_REPO_ID:-vaibhavkhandare/train-bhai-train}"
 NB_EXEC_TIMEOUT="${NB_EXEC_TIMEOUT:-3600}"