From 3494cb006731af4f483c607e6c493fa096357c40 Mon Sep 17 00:00:00 2001 From: Nicolas Patry Date: Mon, 6 May 2024 20:16:15 +0200 Subject: [PATCH] Layernorm import. --- server/text_generation_server/layers/__init__.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/server/text_generation_server/layers/__init__.py b/server/text_generation_server/layers/__init__.py index d0469088..c727c1c7 100644 --- a/server/text_generation_server/layers/__init__.py +++ b/server/text_generation_server/layers/__init__.py @@ -8,3 +8,7 @@ from text_generation_server.layers.linear import ( get_linear, FastLinear, ) +from text_generation_server.layers.layernorm import ( + get_linear, + FastLinear, +)