Skip to content

Commit

Permalink
updated orca pf benchmarks
Browse files Browse the repository at this point in the history
Signed-off-by: 1000850000 user <[email protected]>
  • Loading branch information
achew010 committed Sep 5, 2024
1 parent fe9c3d1 commit f54550d
Show file tree
Hide file tree
Showing 2 changed files with 88 additions and 8 deletions.
88 changes: 88 additions & 0 deletions scripts/benchmarks/refs_orca/requirements_pf.txt
Original file line number Diff line number Diff line change
@@ -0,0 +1,88 @@
accelerate==0.33.0
aiohappyeyeballs==2.4.0
aiohttp==3.10.5
aiosignal==1.3.1
async-timeout==4.0.3
attrs==24.2.0
bitsandbytes==0.43.3
certifi==2024.8.30
charset-normalizer==3.3.2
contourpy==1.3.0
cycler==0.12.1
datasets==2.21.0
dill==0.3.8
docstring_parser==0.16
einops==0.8.0
filelock==3.15.4
fire==0.6.0
flash-attn==2.6.3
-e git+https://github.com/foundation-model-stack/fms-acceleration.git@4831b35b3f2d8c8116e55faed3d050a51ef962a0#egg=fms_acceleration&subdirectory=plugins/framework
-e git+https://github.com/foundation-model-stack/fms-acceleration.git@4831b35b3f2d8c8116e55faed3d050a51ef962a0#egg=fms_acceleration_aadp&subdirectory=plugins/attention-and-distributed-packing
-e git+https://github.com/foundation-model-stack/fms-acceleration.git@4831b35b3f2d8c8116e55faed3d050a51ef962a0#egg=fms_acceleration_foak&subdirectory=plugins/fused-ops-and-kernels
-e git+https://github.com/foundation-model-stack/fms-acceleration.git@4831b35b3f2d8c8116e55faed3d050a51ef962a0#egg=fms_acceleration_peft&subdirectory=plugins/accelerated-peft
fms-hf-tuning @ git+https://github.com/foundation-model-stack/fms-hf-tuning.git@05b40037a3876c2001dda8535dfa3f9b83714b8e
fonttools==4.53.1
frozenlist==1.4.1
fsspec==2024.6.1
huggingface-hub==0.24.6
idna==3.8
Jinja2==3.1.4
kiwisolver==1.4.5
llvmlite==0.43.0
markdown-it-py==3.0.0
MarkupSafe==2.1.5
matplotlib==3.9.2
mdurl==0.1.2
mpmath==1.3.0
multidict==6.0.5
multiprocess==0.70.16
networkx==3.3
numba==0.60.0
numpy==1.26.4
nvidia-cublas-cu12==12.1.3.1
nvidia-cuda-cupti-cu12==12.1.105
nvidia-cuda-nvrtc-cu12==12.1.105
nvidia-cuda-runtime-cu12==12.1.105
nvidia-cudnn-cu12==9.1.0.70
nvidia-cufft-cu12==11.0.2.54
nvidia-curand-cu12==10.3.2.106
nvidia-cusolver-cu12==11.4.5.107
nvidia-cusparse-cu12==12.1.0.106
nvidia-nccl-cu12==2.20.5
nvidia-nvjitlink-cu12==12.6.68
nvidia-nvtx-cu12==12.1.105
packaging==24.1
pandas==2.2.2
peft==0.12.0
pillow==10.4.0
protobuf==5.28.0
psutil==6.0.0
pyarrow==17.0.0
Pygments==2.18.0
pyparsing==3.1.4
python-dateutil==2.9.0.post0
pytz==2024.1
PyYAML==6.0.2
regex==2024.7.24
requests==2.32.3
rich==13.8.0
safetensors==0.4.4
sentencepiece==0.2.0
shtab==1.7.1
simpleeval==0.9.13
six==1.16.0
sympy==1.13.2
termcolor==2.4.0
threadpoolctl==3.5.0
tokenizers==0.19.1
torch==2.4.0
tqdm==4.66.5
transformers==4.44.2
triton==3.0.0
trl==0.10.1
typing_extensions==4.12.2
tyro==0.8.10
tzdata==2024.1
urllib3==2.2.2
xxhash==3.5.0
yarl==1.9.7
8 changes: 0 additions & 8 deletions scripts/benchmarks/scenarios-orca.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -49,8 +49,6 @@ scenarios:
packing: False
model_name_or_path:
- 'mistralai/Mistral-7B-v0.1'
# dataset_text_field: null
# response_template: null
dataset_text_field: 'output'
response_template: "\n\nASSISTANT:"

Expand All @@ -66,8 +64,6 @@ scenarios:
packing: False
model_name_or_path:
- 'mistralai/Mistral-7B-v0.1'
# dataset_text_field: null
# response_template: null
dataset_text_field: 'output'
response_template: "\n\nASSISTANT:"

Expand All @@ -91,8 +87,6 @@ scenarios:
packing: False
model_name_or_path:
- 'mistralai/Mistral-7B-v0.1'
# dataset_text_field: null
# response_template: null
dataset_text_field: 'output'
response_template: "\n\nASSISTANT:"

Expand All @@ -116,7 +110,5 @@ scenarios:
packing: False
model_name_or_path:
- 'TheBloke/Mistral-7B-v0.1-GPTQ'
# dataset_text_field: null
# response_template: null
dataset_text_field: 'output'
response_template: "\n\nASSISTANT:"

0 comments on commit f54550d

Please sign in to comment.