feat: 测试脚本挂载路径更新

520f3c9f · chenpangpang · 66b363fa · 520f3c9f · 520f3c9f
Commit 520f3c9f authored Sep 18, 2024 by chenpangpang
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 2 deletions

script/2_text_generate_test.sh script/2_text_generate_test.sh +1 -1

script/3_image_generate_test.sh script/3_image_generate_test.sh +1 -1

No files found.
--- a/script/2_text_generate_test.sh
+++ b/script/2_text_generate_test.sh
 #!/bin/bash
-
+TARGET_DIR=gpu-base-image-test
 docker run --rm --platform=linux/amd64 --gpus all -v ./$TARGET_DIR:/workspace --workdir /workspace/gpt2 $1 python infer.py
\ No newline at end of file
--- a/script/3_image_generate_test.sh
+++ b/script/3_image_generate_test.sh
 #!/bin/bash
-
+TARGET_DIR=gpu-base-image-test
 docker run --rm --platform=linux/amd64 --gpus all -v ./$TARGET_DIR:/workspace --workdir /workspace/stable-diffusion-v1-4 $1 python infer.py
\ No newline at end of file