Add accelerate and diffusers to setup.py

nod-ai · Dec 6, 2023 · 49d1dfb · 49d1dfb
1 parent 40ee79b
commit 49d1dfb
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/python/turbine_models/setup.py b/python/turbine_models/setup.py
@@ -63,5 +63,7 @@ def load_version_info():
         "protobuf",
         "sentencepiece",
         "transformers",
+        "accelerate",
+        "diffusers==0.10.2",
     ],
 )