Skip to content

Commit

Permalink
take host env in run-benches
Browse files Browse the repository at this point in the history
  • Loading branch information
fabianlim committed May 17, 2024
1 parent 108a639 commit 3635e7e
Showing 1 changed file with 1 addition and 5 deletions.
6 changes: 1 addition & 5 deletions tox.ini
Original file line number Diff line number Diff line change
Expand Up @@ -23,11 +23,7 @@ skip_install = true
deps =
packaging # this is required for flash-attn dep as fms_hf_tuning did not specify
-e {toxinidir}/plugins/framework # install the framework here as the flash attention deps requires torch
setenv =
FHT_BRANCH = {env:FHT_BRANCH:main}
DRY_RUN = {env:DRY_RUN:false}
NO_DATA_PROCESSING = {env:NO_DATA_PROCESSING:false}
NO_OVERWRITE = {env:NO_OVERWRITE:false}
passenv = * # will pass the parent env, otherwise there are too many envs e.g. TRANSFORMERS that need to be set
commands =
# need a version of fms-hf-tuning that has integrated the framework
# NOTE: have to install this first coz havnt merged
Expand Down

0 comments on commit 3635e7e

Please sign in to comment.