Ver Fonte

Fix transformers version

Alexander Borzunov há 3 anos atrás
pai
commit
3db71b9de5
1 ficheiros alterados com 2 adições e 2 exclusões
  1. 2 2
      requirements.txt

+ 2 - 2
requirements.txt

@@ -1,7 +1,7 @@
 git+git://github.com/learning-at-home/hivemind@dalle-v1
 git+git://github.com/learning-at-home/dalle-pytorch@weight-sharing
 bitsandbytes-cuda111>=0.26.0
-transformers>=4.9.2
+transformers==4.12.2
 tokenizers>=0.10.2
 datasets>=1.11.0
 torch_optimizer>=0.1.0
@@ -10,4 +10,4 @@ nltk>=3.6.2
 sentencepiece
 aiohttp
 requests>=2.24.0
-termcolor
+termcolor