diff --git a/examples/quantization/requirements.txt b/examples/quantization/requirements.txt index 21e5b8362..ce63abe54 100644 --- a/examples/quantization/requirements.txt +++ b/examples/quantization/requirements.txt @@ -3,3 +3,4 @@ nemo-toolkit[all]<=1.20.0,>=1.18.0 rouge_score~=0.1.2 transformers_stream_generator==0.0.4 tiktoken +mpmath==1.3.0