Skip to content

Commit 24150d2

Browse files
committed
readme updates for new config file changes, docs updates
1 parent 1f2b739 commit 24150d2

File tree

7 files changed

+44
-1
lines changed

7 files changed

+44
-1
lines changed

README.md

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -88,6 +88,11 @@ Llama3 is now available on SageMaker (read [blog post](https://aws.amazon.com/bl
8888

8989
## New in this release
9090

91+
## 2.0.26
92+
93+
1. Bug fix for missing HuggingFace token file.
94+
1. Config file enhancements
95+
9196
## 2.0.25
9297

9398
1. Fix bug with an alternate VariantName for SageMaker BYOE.

docs/build.md

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -10,6 +10,7 @@ If you would like to build a dev version of `FMBench` for your own development a
1010

1111
```{.bash}
1212
pip install poetry mkdocs-material mknotebooks
13+
sudo apt-get install tree
1314
```
1415
1516
1. Change directory to the `FMBench` repo directory and run poetry build.

docs/cli.md

Lines changed: 31 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,31 @@
1+
# `FMBench` CLI
2+
3+
Here are the command line options currently supported by the `fmbench` CLI.
4+
5+
```{.bashrc}
6+
usage: fmbench [-h] --config-file CONFIG_FILE [--role-arn ROLE_ARN] [--local-mode {yes,no}] [--tmp-dir TMP_DIR] [--write-bucket WRITE_BUCKET] -A [key=value]
7+
8+
Run FMBench with a specified config file.
9+
```
10+
11+
options:
12+
`-h`, `--help` show this help message and exit
13+
14+
`--config-file` CONFIG_FILE
15+
The S3 URI of your Config File
16+
17+
`--role-arn` ROLE_ARN (_Optional_) The ARN of the role to be used for FMBench. If an Amazon SageMaker endpoint is being deployed through FMBench then this role would also be used by that endpoint
18+
19+
`--local-mode` {yes,no} Specify if running in local mode or not. Options: yes, no. Default is no.
20+
21+
`--tmp-dir` TMP_DIR (_Optional_) An optional tmp directory if fmbench is running in local mode.
22+
23+
`--write-bucket` WRITE_BUCKET Write bucket that is used for sagemaker endpoints in local mode and storing metrics in s3 mode.
24+
25+
`-A` key=value Specify a key value pair which will be used to replace the {key} in the given config file. The key could be anything that you have templatized
26+
in the config file as `{key}` and it will be replaced with `value`. This comes in handy when using a generic configuration file and replace keys such `model_id`,
27+
`tp_degree` etc. Note that you are not limited to pre-defined set of keys, you can put any key in the config file as `{key}` and it will get replaced with its value.
28+
If there are multiple key value pairs, then simply specificy the `-A` option multiple times in the command line.
29+
30+
```
31+

docs/manifest.md

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -39,6 +39,10 @@ Here is a listing of the various configuration files available out-of-the-box wi
3939
[└── embeddings/bge-base-en-v1-5-g5-g4dn-c7-embeddings.yml](configs/embeddings/bge-base-en-v1-5-g5-g4dn-c7-embeddings.yml)
4040
**gemma**
4141
[└── gemma/config-gemma-2b-g5.yml](configs/gemma/config-gemma-2b-g5.yml)
42+
**generic**
43+
**└── generic/ec2**
44+
[├── generic/ec2/djl.yml](configs/generic/ec2/djl.yml)
45+
[└── generic/ec2/llama3.1_djl.yml](configs/generic/ec2/llama3.1_djl.yml)
4246
**llama2**
4347
**├── llama2/13b**
4448
[│   ├── llama2/13b/config-bedrock-sagemaker-llama2.yml](configs/llama2/13b/config-bedrock-sagemaker-llama2.yml)

manifest.txt

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -28,6 +28,8 @@ configs/embeddings/bge-base-en-v1-5-c5-embeddings.yml
2828
configs/embeddings/bge-base-en-v1-5-g5-embeddings.yml
2929
configs/embeddings/bge-base-en-v1-5-g5-g4dn-c7-embeddings.yml
3030
configs/gemma/config-gemma-2b-g5.yml
31+
configs/generic/ec2/djl.yml
32+
configs/generic/ec2/llama3.1_djl.yml
3133
configs/llama2/13b/config-bedrock-sagemaker-llama2.yml
3234
configs/llama2/13b/config-byo-rest-ep-llama2-13b.yml
3335
configs/llama2/13b/config-llama2-13b-inf2-g5-p4d.yml

pyproject.toml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,6 @@
11
[tool.poetry]
22
name = "fmbench"
3-
version = "2.0.25"
3+
version = "2.0.26"
44
description ="Benchmark performance of **any Foundation Model (FM)** deployed on **any AWS Generative AI service**, be it **Amazon SageMaker**, **Amazon Bedrock**, **Amazon EKS**, or **Amazon EC2**. The FMs could be deployed on these platforms either directly through `FMbench`, or, if they are already deployed then also they could be benchmarked through the **Bring your own endpoint** mode supported by `FMBench`."
55
authors = ["Amit Arora <[email protected]>", "Madhur Prashant <[email protected]>"]
66
readme = "README.md"

0 commit comments

Comments
 (0)