remove import

This commit is contained in:
Mohit Sharma 2024-09-27 12:59:35 +00:00
parent a24c2cc5e9
commit 346dfe398a

View File

@ -19,8 +19,6 @@ from text_generation_server.layers.lora import (
TensorParallelAdapterRowLinear, TensorParallelAdapterRowLinear,
) )
from text_generation_server.layers.moe.fused_moe_rocm import grouped_topk
__all__ = [ __all__ = [
"get_linear", "get_linear",
"FastLinear", "FastLinear",
@ -33,5 +31,4 @@ __all__ = [
"TensorParallelAdapterRowLinear", "TensorParallelAdapterRowLinear",
"load_layer_norm", "load_layer_norm",
"load_conv2d", "load_conv2d",
"grouped_topk",
] ]