Georg commited on
Commit
8d70077
·
1 Parent(s): 738da22

Update base image build and deps

Browse files
Files changed (2) hide show
  1. deploy.sh +1 -0
  2. scripts/run_hf_image_job.py +2 -2
deploy.sh CHANGED
@@ -52,6 +52,7 @@ JOB_OUTPUT=$("${PY_BIN}" scripts/run_hf_image_job.py \
52
  --platform "${PLATFORM}" \
53
  --dockerfile "Dockerfile.base" \
54
  --target "foundationpose-base-l2" \
 
55
  --git-repo "https://huggingface.co/spaces/${HF_SPACE}" 2>&1 | tee /tmp/hf_image_job.log)
56
 
57
  JOB_ID=$(echo "${JOB_OUTPUT}" | awk '/Job ID:/ {print $3}')
 
52
  --platform "${PLATFORM}" \
53
  --dockerfile "Dockerfile.base" \
54
  --target "foundationpose-base-l2" \
55
+ --flavor "l40sx1" \
56
  --git-repo "https://huggingface.co/spaces/${HF_SPACE}" 2>&1 | tee /tmp/hf_image_job.log)
57
 
58
  JOB_ID=$(echo "${JOB_OUTPUT}" | awk '/Job ID:/ {print $3}')
scripts/run_hf_image_job.py CHANGED
@@ -51,8 +51,8 @@ def main() -> None:
51
  )
52
  parser.add_argument(
53
  "--flavor",
54
- default="l40s",
55
- help="HF Jobs hardware flavor (default: l40s)",
56
  )
57
  parser.add_argument(
58
  "--timeout",
 
51
  )
52
  parser.add_argument(
53
  "--flavor",
54
+ default="l40sx1",
55
+ help="HF Jobs hardware flavor (default: l40sx1)",
56
  )
57
  parser.add_argument(
58
  "--timeout",