Add HF hub as a dependency (#142)

mosaicml · May 16, 2024 · 46446f5 · 46446f5
1 parent 35fa48a
commit 46446f5
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/setup.py b/setup.py
@@ -12,6 +12,7 @@
     'hydra-colorlog>=1.1.0',
     'diffusers[torch]==0.26.3',
     'transformers[torch]==4.38.2',
+    'huggingface_hub==0.21.2',
     'wandb==0.16.3',
     'xformers==0.0.23.post1',
     'triton==2.1.0',