Add accelerate to dependencies

frankier · Apr 30, 2024 · 798aa88 · 798aa88
1 parent 62fa709
commit 798aa88
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/environment.yml b/environment.yml
@@ -35,6 +35,8 @@ dependencies:
   - jupyterlab-lsp
   - jupyter-resource-usage
   - sentence-transformers
+  - transformers
+  - accelerate
   - llama-cpp-python
   - otter-grader
   - tabulate