Download 14B/scripts/infer.sh from Elastic-forcing/Elastic-forcing-code: direct link, hf CLI and curl.
- Browser
- Download file 665 Bytes
-
https://huggingface.co/Elastic-forcing/Elastic-forcing-code/resolve/main/14B/scripts/infer.sh
- Command line
-
hf download hf://Elastic-forcing/Elastic-forcing-code/14B/scripts/infer.sh
-
curl -L -o infer.sh https://huggingface.co/Elastic-forcing/Elastic-forcing-code/resolve/main/14B/scripts/infer.sh
665 Bytes
| set -euo pipefail | |
| source "$(dirname "${BASH_SOURCE[0]}")/common.sh" | |
| : "${CHECKPOINT:?Set CHECKPOINT to the step-80 model.pt file}" | |
| CHECKPOINT="$(realpath "$CHECKPOINT")" | |
| PROMPTS="$(realpath "${PROMPTS:-$REPRO_ROOT/code/prompts/test_8.txt}")" | |
| OUTPUT_DIR="$(realpath -m "${OUTPUT_DIR:-$REPRO_ROOT/outputs/demo}")" | |
| export CUDA_VISIBLE_DEVICES="${GPUS:-0}" | |
| cd "$REPRO_ROOT/code" | |
| "$PYTHON_BIN" inference.py --config_path "$REPRO_ROOT/configs/inference.yaml" \ | |
| --checkpoint_path "$CHECKPOINT" --data_path "$PROMPTS" \ | |
| --output_folder "$OUTPUT_DIR" --num_output_frames 21 \ | |
| --num_samples "${NUM_SAMPLES:-5}" --seed "${SEED:-0}" --save_with_index | |