diff --git a/examples/tutorial/opt/inference/requirements.txt b/examples/tutorial/opt/inference/requirements.txt index e6e8511e3..966dff474 100644 --- a/examples/tutorial/opt/inference/requirements.txt +++ b/examples/tutorial/opt/inference/requirements.txt @@ -7,3 +7,4 @@ torch>=1.10.0 transformers==4.23.1 uvicorn==0.19.0 colossalai +git+https://github.com/hpcaitech/EnergonAI@main