nodes/hao-ai-lab-FastVideo
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix

Browse Source
This commit is contained in:
RandNMR73
2026-04-09 19:36:53 +00:00
parent 15bda1970f
commit b978a92ef3
1 changed files with 19 additions and 40 deletions
Download Patch File Download Diff File Expand all files Collapse all files
.buildkite/scripts/pr_test.sh
+19 -40
View File
@@ -39,11 +39,7 @@ if [ -n "$MODAL_TOKEN_ID" ] && [ -n "$MODAL_TOKEN_SECRET" ]; then
log "Retrieved Modal credentials from Buildkite secrets"
python3 -m modal token set --token-id "$MODAL_TOKEN_ID" --token-secret "$MODAL_TOKEN_SECRET" --profile buildkite-ci --activate --verify
if [ $? -eq 0 ]; then
export MODAL_PROFILE=buildkite-ci
export MODAL_TOKEN_ID
export MODAL_TOKEN_SECRET
log "Modal authentication successful"
log "Using Modal profile: $MODAL_PROFILE"
else
log "Error: Failed to set Modal credentials"
exit 1
@@ -67,90 +63,73 @@ EFFECTIVE_PR=${BUILDKITE_PULL_REQUEST:-false}
if [ "$EFFECTIVE_PR" = "false" ] && [ -n "${PR_NUMBER:-}" ]; then
EFFECTIVE_PR=$PR_NUMBER
fi
export BUILDKITE_REPO
export BUILDKITE_COMMIT
export BUILDKITE_PULL_REQUEST=$EFFECTIVE_PR
export IMAGE_VERSION
MODAL_TARGET=""
MODAL_ENV="BUILDKITE_REPO=$BUILDKITE_REPO BUILDKITE_COMMIT=$BUILDKITE_COMMIT BUILDKITE_PULL_REQUEST=$EFFECTIVE_PR IMAGE_VERSION=$IMAGE_VERSION"
case "$TEST_TYPE" in
"encoder")
log "Running encoder tests..."
export HF_API_KEY
MODAL_TARGET="$MODAL_TEST_FILE::run_encoder_tests"
MODAL_COMMAND="$MODAL_ENV HF_API_KEY=$HF_API_KEY python3 -m modal run $MODAL_TEST_FILE::run_encoder_tests"
;;
"vae")
log "Running VAE tests..."
export HF_API_KEY
MODAL_TARGET="$MODAL_TEST_FILE::run_vae_tests"
MODAL_COMMAND="$MODAL_ENV HF_API_KEY=$HF_API_KEY python3 -m modal run $MODAL_TEST_FILE::run_vae_tests"
;;
"transformer")
log "Running transformer tests..."
export HF_API_KEY
MODAL_TARGET="$MODAL_TEST_FILE::run_transformer_tests"
MODAL_COMMAND="$MODAL_ENV HF_API_KEY=$HF_API_KEY python3 -m modal run $MODAL_TEST_FILE::run_transformer_tests"
;;
"ssim")
log "Running SSIM tests..."
export HF_API_KEY
MODAL_TARGET="$MODAL_SSIM_TEST_FILE::run_ssim_tests"
MODAL_COMMAND="$MODAL_ENV HF_API_KEY=$HF_API_KEY python3 -m modal run $MODAL_SSIM_TEST_FILE::run_ssim_tests"
;;
"training")
log "Running training tests..."
export WANDB_API_KEY
MODAL_TARGET="$MODAL_TEST_FILE::run_training_tests"
MODAL_COMMAND="$MODAL_ENV WANDB_API_KEY=$WANDB_API_KEY python3 -m modal run $MODAL_TEST_FILE::run_training_tests"
;;
"training_lora")
log "Running LoRA training tests..."
export WANDB_API_KEY
MODAL_TARGET="$MODAL_TEST_FILE::run_training_lora_tests"
MODAL_COMMAND="$MODAL_ENV WANDB_API_KEY=$WANDB_API_KEY python3 -m modal run $MODAL_TEST_FILE::run_training_lora_tests"
;;
"training_vsa")
log "Running training VSA tests..."
export WANDB_API_KEY
MODAL_TARGET="$MODAL_TEST_FILE::run_training_tests_VSA"
MODAL_COMMAND="$MODAL_ENV WANDB_API_KEY=$WANDB_API_KEY python3 -m modal run $MODAL_TEST_FILE::run_training_tests_VSA"
;;
"kernel_tests")
log "Running kernel tests..."
MODAL_TARGET="$MODAL_TEST_FILE::run_kernel_tests"
MODAL_COMMAND="$MODAL_ENV python3 -m modal run $MODAL_TEST_FILE::run_kernel_tests"
;;
"inference_lora")
log "Running LoRA tests..."
MODAL_TARGET="$MODAL_TEST_FILE::run_inference_lora_tests"
MODAL_COMMAND="$MODAL_ENV python3 -m modal run $MODAL_TEST_FILE::run_inference_lora_tests"
;;
"distillation_dmd")
log "Running distillation DMD tests..."
export WANDB_API_KEY
MODAL_TARGET="$MODAL_TEST_FILE::run_distill_dmd_tests"
MODAL_COMMAND="$MODAL_ENV WANDB_API_KEY=$WANDB_API_KEY python3 -m modal run $MODAL_TEST_FILE::run_distill_dmd_tests"
;;
# run_inference_tests_vmoba
"self_forcing")
log "Running self-forcing tests..."
export WANDB_API_KEY
MODAL_TARGET="$MODAL_TEST_FILE::run_self_forcing_tests"
MODAL_COMMAND="$MODAL_ENV WANDB_API_KEY=$WANDB_API_KEY python3 -m modal run $MODAL_TEST_FILE::run_self_forcing_tests"
;;
"inference_vmoba")
log "Running V-MoBA inference tests..."
MODAL_TARGET="$MODAL_TEST_FILE::run_inference_tests_vmoba"
MODAL_COMMAND="$MODAL_ENV python3 -m modal run $MODAL_TEST_FILE::run_inference_tests_vmoba"
;;
"unit_test")
log "Running unit tests..."
MODAL_TARGET="$MODAL_TEST_FILE::run_unit_test"
MODAL_COMMAND="$MODAL_ENV python3 -m modal run $MODAL_TEST_FILE::run_unit_test"
;;
"lora_extraction")
log "Running LoRA extraction tests..."
export HF_API_KEY
MODAL_TARGET="$MODAL_TEST_FILE::run_lora_extraction_tests"
MODAL_COMMAND="$MODAL_ENV HF_API_KEY=$HF_API_KEY python3 -m modal run $MODAL_TEST_FILE::run_lora_extraction_tests"
;;
"performance")
log "Running performance tests..."
export HF_API_KEY
MODAL_TARGET="$MODAL_TEST_FILE::run_performance_tests"
MODAL_COMMAND="$MODAL_ENV HF_API_KEY=$HF_API_KEY python3 -m modal run $MODAL_TEST_FILE::run_performance_tests"
;;
"api_server")
log "Running API server integration tests..."
export HF_API_KEY
MODAL_TARGET="$MODAL_TEST_FILE::run_api_server_tests"
MODAL_COMMAND="$MODAL_ENV HF_API_KEY=$HF_API_KEY python3 -m modal run $MODAL_TEST_FILE::run_api_server_tests"
;;
*)
log "Error: Unknown test type: $TEST_TYPE"
@@ -158,8 +137,8 @@ case "$TEST_TYPE" in
;;
esac
log "Executing: python3 -m modal run $MODAL_TARGET"
python3 -m modal run "$MODAL_TARGET"
log "Executing: $MODAL_COMMAND"
eval "$MODAL_COMMAND"
TEST_EXIT_CODE=$?
if [ $TEST_EXIT_CODE -eq 0 ]; then