AuraFlow model implementation.

2024-07-11 16:51:06 -04:00
parent f45157e3ac
commit 9f291d75b3
12 changed files with 1744 additions and 2 deletions
--- a/requirements.txt
+++ b/requirements.txt
@@ -3,7 +3,8 @@ torchsde
 torchvision
 torchaudio
 einops
-transformers>=4.25.1
+transformers>=4.28.1
+tokenizers>=0.13.3
 safetensors>=0.4.2
 aiohttp
 pyyaml