Add timm to setup.py

huggingface · Jun 27, 2023 · a25e2d5 · a25e2d5
1 parent d7a087a
commit a25e2d5
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/setup.py b/setup.py
@@ -33,6 +33,7 @@
     "torchvision",
     "diffusers>=0.17.0",
     "torchaudio",
+    "timm"
 ]
 
 QUALITY_REQUIRE = ["black~=23.1", "ruff>=0.0.241,<=0.0.259"]