diff --git a/nemoguardrails/library/jailbreak_detection/requirements.txt b/nemoguardrails/library/jailbreak_detection/requirements.txt index d970b083f..cbe020128 100644 --- a/nemoguardrails/library/jailbreak_detection/requirements.txt +++ b/nemoguardrails/library/jailbreak_detection/requirements.txt @@ -7,5 +7,6 @@ uvicorn>=0.23.2 transformers>=4.32.1 torch>=2.1.1 nemoguardrails>=0.7.0 +numpy==1.23.5 scikit-learn==1.2.2 einops>=0.7.0