mirror of
https://github.com/huggingface/text-generation-inference.git
synced 2025-05-13 05:22:09 +00:00
24 lines
857 B
Makefile
24 lines
857 B
Makefile
.PHONY: download-dataset load-test build-k6
|
|
download-dataset:
|
|
@if [ ! -f ./benchmarks/ShareGPT_V3_unfiltered_cleaned_split.json ]; then \
|
|
echo "Downloading dataset"; \
|
|
curl -L -o benchmarks/ShareGPT_V3_unfiltered_cleaned_split.json https://huggingface.co/datasets/anon8231489123/ShareGPT_Vicuna_unfiltered/resolve/main/ShareGPT_V3_unfiltered_cleaned_split.json; \
|
|
else \
|
|
echo "Dataset already downloaded"; \
|
|
fi
|
|
|
|
load-test: download-dataset build-k6
|
|
poetry install && poetry run python load_test.py
|
|
|
|
build-k6:
|
|
mkdir -p /tmp/xk6 && \
|
|
cd /tmp/xk6 && \
|
|
git clone https://github.com/mstoykov/xk6-sse.git && \
|
|
cd xk6-sse && \
|
|
git checkout useSobek && \
|
|
go install go.k6.io/xk6/cmd/xk6@latest && \
|
|
xk6 build --with github.com/phymbert/xk6-sse=. && \
|
|
mkdir -p ~/.local/bin/ && \
|
|
mv k6 /tmp/k6-sse && \
|
|
rm -rf /tmp/xk6 && \
|
|
/tmp/k6-sse --version
|