remove import

huggingface · Sep 27, 2024 · 346dfe3 · 346dfe3
1 parent a24c2cc
commit 346dfe3
Showing 1 changed file with 0 additions and 3 deletions.
diff --git a/server/text_generation_server/layers/__init__.py b/server/text_generation_server/layers/__init__.py
@@ -19,8 +19,6 @@
     TensorParallelAdapterRowLinear,
 )
 
-from text_generation_server.layers.moe.fused_moe_rocm import grouped_topk
-
 __all__ = [
     "get_linear",
     "FastLinear",
@@ -33,5 +31,4 @@
     "TensorParallelAdapterRowLinear",
     "load_layer_norm",
     "load_conv2d",
-    "grouped_topk",
 ]