add packaging dep

Signed-off-by: Yu Chin Fabian Lim <flim@sg.ibm.com>
foundation-model-stack · May 17, 2024 · 0aa4437 · 0aa4437
1 parent 697bbca
commit 0aa4437
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 0 deletions.
diff --git a/scripts/benchmarks/README.md b/scripts/benchmarks/README.md
@@ -51,6 +51,11 @@ A `scenario` has the following key components:
 
 The best way is via `tox` which manages the dependencies, including installing the correct version [fms-hf-tuning](https://github.com/foundation-model-stack/fms-hf-tuning).
 
+- install the `setup_requirements.txt` to get `tox`:
+    ```
+    pip install -r setup_requirements.txt
+    ```
+
 - run a *small* representative set of benches:
     ```
     tox -e run-benches

diff --git a/tox.ini b/tox.ini
@@ -20,6 +20,7 @@ allowlist_externals = bash
 [testenv:run-benches]
 description = run benchmarks 
 skip_install = true
+deps = packaging # this is required for flash-attn dep as fms_hf_tuning did not specify
 setenv = 
     FHT_BRANCH = {env:FHT_BRANCH:main}
     DRY_RUN = {env:DRY_RUN:false}