This commit is contained in:
Nicolas Patry 2024-05-07 13:19:49 +00:00
parent cc07a84604
commit 72f5282098

View File

@ -11,3 +11,4 @@ from text_generation_server.layers.linear import (
# Just to add the `load` methods.
from text_generation_server.layers.layernorm import load_layer_norm
from text_generation_server.layers.conv import load_conv2d