Skip to content

Commit ae98dc5

Browse files
committed
add generation dataset to docker compose
1 parent 4fa15b8 commit ae98dc5

File tree

2 files changed

+5
-1
lines changed

2 files changed

+5
-1
lines changed
Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,3 @@
1+
## How to run the docker compose file ##
2+
3+
```UID=$(id -u) GID=$(id -g) docker compose -f docker-compose-qwen.yml up```

language/qwen2.5-VL-7B-docker/docker-compose-qwen.yml

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -77,6 +77,7 @@ services:
7777
echo 'vLLM server is healthy, starting script...'
7878
pip install -r /app/requirements-script.txt
7979
pip install /loadgen/
80+
cd datasets && python generate_json_dataset.py && cd ../ &&
8081
python alt_main.py --scenario Server \
8182
--model-path $${CHECKPOINT_PATH} \
8283
--batch-size 16 \
@@ -87,7 +88,7 @@ services:
8788
--dataset-path $${DATASET_PATH} \
8889
--output-log-dir output \
8990
--tensor-parallel-size $${GPU_COUNT} \
90-
--num-workers 8 \
91+
--num-workers 16 \
9192
--vllm
9293
9394
networks:

0 commit comments

Comments
 (0)