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

fix dbrx & opt model prefix bug

Browse Source
This commit is contained in:
roc 2024-07-08 14:47:51 +08:00
parent 28cecb663a
commit 39944e1cbb
2 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
server/text_generation_server/models/custom_modeling/flash_dbrx_modeling.py
View File

@ -711,7 +711,7 @@ class FlashDbrxForCausalLM(torch.nn.Module):
else:
prefix = f"{prefix}.transformer"
self.model = DbrxModel(config, weights)
self.model = DbrxModel(prefix, config, weights)
self.lm_head = SpeculativeHead.load(
config,
prefix="lm_head",

2
server/text_generation_server/models/custom_modeling/opt_modeling.py
View File

@ -757,7 +757,7 @@ class OPTForCausalLM(OPTPreTrainedModel):
else:
prefix = f"{prefix}.model"
self.model = OPTModel(config, weights)
self.model = OPTModel(prefix, config, weights)
self.lm_head = SpeculativeHead.load(
config, prefix=f"{prefix}.decoder.embed_tokens", weights=weights