[NPU] [Bug Fix] Fix typo in npu device check in gpt_oss.py (#17553)
This commit is contained in:
@@ -492,7 +492,7 @@ class GptOssModel(nn.Module):
|
|||||||
self.vocab_size = config.vocab_size
|
self.vocab_size = config.vocab_size
|
||||||
self.pp_group = get_pp_group()
|
self.pp_group = get_pp_group()
|
||||||
|
|
||||||
if is_npu:
|
if _is_npu:
|
||||||
config.hidden_act = "npu_swiglu_oai"
|
config.hidden_act = "npu_swiglu_oai"
|
||||||
|
|
||||||
if self.pp_group.is_first_rank:
|
if self.pp_group.is_first_rank:
|
||||||
|
|||||||
Reference in New Issue
Block a user