huggingface/text-generation-inference
5
0
Fork 0
mirror of https://github.com/huggingface/text-generation-inference.git synced 2025-09-10 03:44:54 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix for new localization.

Browse Source
This commit is contained in:
Nicolas Patry 2023-05-15 10:59:49 +02:00
parent 62b4082514
commit f64c9ba305
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
server/text_generation_server/models/model.py
View File

@ -103,7 +103,7 @@ class Model(ABC):
def check_initialized(self): def check_initialized(self):
uninitialized_parameters = [] uninitialized_parameters = []
for n, p in self.named_parameters(): for n, p in self.model.named_parameters():
if p.data.device == torch.device("meta"): if p.data.device == torch.device("meta"):
uninitialized_parameters.append(n) uninitialized_parameters.append(n)
if uninitialized_parameters: if uninitialized_parameters: