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