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

formatting

Browse Source
This commit is contained in:
OlivierDehaene 2023-01-31 17:38:12 +01:00
parent ca11e9e8c3
commit 4858f122db
2 changed files with 2 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
server/text_generation/models/bloom.py
View File

@ -217,9 +217,7 @@ class BLOOMSharded(BLOOM):
return linear
module.linear = replace_linear(
state
)
module.linear = replace_linear(state)
else:
tensor = tensor.to(device)

4
server/text_generation/models/galactica.py
View File

@ -315,9 +315,7 @@ class GalacticaSharded(Galactica):
return linear
module.linear = replace_linear(
state
)
module.linear = replace_linear(state)
else:
tensor = tensor.to(device)