diff --git a/environment.yml b/environment.yml index e476c8b..eed37f4 100644 --- a/environment.yml +++ b/environment.yml @@ -20,4 +20,5 @@ dependencies: - pip - pip: - git+https://github.com/igdrion/transforkmers.git - - accelerate=0.24.1 + - accelerate==0.24.1 + - transformers==4.35.2