Extra serverless (#1320)

* update * update * Move back * update * update * rename --------- Co-authored-by: Xiaohan Zhang <[email protected]>
mosaicml · Jul 1, 2024 · 6117844 · 6117844
1 parent 68c2625
commit 6117844
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/setup.py b/setup.py
@@ -123,6 +123,11 @@
     'grouped-gemm==0.1.4',
 ]
 
+extra_deps['databricks-serverless'] = {
+    dep for key, deps in extra_deps.items() for dep in deps
+    if 'gpu' not in key and 'megablocks' not in key and
+    'databricks-connect' not in dep
+}
 extra_deps['all-cpu'] = {
     dep for key, deps in extra_deps.items() for dep in deps
     if 'gpu' not in key and 'megablocks' not in key