diff --git a/environment.yml b/environment.yml index 98e72d50b..57b63a1e7 100644 --- a/environment.yml +++ b/environment.yml @@ -39,6 +39,7 @@ dependencies: - types-psycopg2 - types-PyYAML - transformers + - nltk - pytorch::pytorch=2.2.1 - pytorch::torchvision - pytorch::cpuonly # comment out if commenting in lines below for CUDA