Spaces:
Sleeping
Sleeping
Georg commited on
Commit ·
8d70077
1
Parent(s): 738da22
Update base image build and deps
Browse files- deploy.sh +1 -0
- 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="
|
| 55 |
-
help="HF Jobs hardware flavor (default:
|
| 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",
|