From 346dfe398af02c7355fe12f2ffbf09223d232ef1 Mon Sep 17 00:00:00 2001
From: Mohit Sharma <mohit21sharma.ms@gmail.com>
Date: Fri, 27 Sep 2024 12:59:35 +0000
Subject: [PATCH] remove import

---
 server/text_generation_server/layers/__init__.py | 3 ---
 1 file changed, 3 deletions(-)

diff --git a/server/text_generation_server/layers/__init__.py b/server/text_generation_server/layers/__init__.py
index e8282b16..0000ca91 100644
--- a/server/text_generation_server/layers/__init__.py
+++ b/server/text_generation_server/layers/__init__.py
@@ -19,8 +19,6 @@ from text_generation_server.layers.lora import (
     TensorParallelAdapterRowLinear,
 )
 
-from text_generation_server.layers.moe.fused_moe_rocm import grouped_topk
-
 __all__ = [
     "get_linear",
     "FastLinear",
@@ -33,5 +31,4 @@ __all__ = [
     "TensorParallelAdapterRowLinear",
     "load_layer_norm",
     "load_conv2d",
-    "grouped_topk",
 ]