From 7fc999b7bdf055af32cf85b02030ef8388ca9660 Mon Sep 17 00:00:00 2001 From: Nicolas Patry Date: Mon, 15 May 2023 15:03:58 +0200 Subject: [PATCH] Update server/text_generation_server/models/flash_neox.py --- server/text_generation_server/models/flash_neox.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/server/text_generation_server/models/flash_neox.py b/server/text_generation_server/models/flash_neox.py index 470fb73b..fc741f55 100644 --- a/server/text_generation_server/models/flash_neox.py +++ b/server/text_generation_server/models/flash_neox.py @@ -150,4 +150,4 @@ class FlashNeoXSharded(FlashNeoX): else: module._buffers[param_name] = tensor - # model.post_load_weights(quantize) + model.post_load_weights(quantize)