From f64c9ba305434076f8689472a29244a1d483d782 Mon Sep 17 00:00:00 2001 From: Nicolas Patry Date: Mon, 15 May 2023 10:59:49 +0200 Subject: [PATCH] Fix for new localization. --- server/text_generation_server/models/model.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/server/text_generation_server/models/model.py b/server/text_generation_server/models/model.py index 31bea6ca..03f14013 100644 --- a/server/text_generation_server/models/model.py +++ b/server/text_generation_server/models/model.py @@ -103,7 +103,7 @@ class Model(ABC): def check_initialized(self): uninitialized_parameters = [] - for n, p in self.named_parameters(): + for n, p in self.model.named_parameters(): if p.data.device == torch.device("meta"): uninitialized_parameters.append(n) if uninitialized_parameters: