ayo/ipex-llm
1
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions 1

Fix ambiguous boolean evaluation in bert.py (#13236)

Browse source 
Signed-off-by: Xue, Zhan <zhan.xue@intel.com>
This commit is contained in:
zxue2 2025-06-30 14:14:01 +08:00 committed by GitHub
parent 3f6d407be4
commit 6ba3138d7c
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 1 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
python/llm/src/ipex_llm/transformers/models/bert.py
View file

@ -119,7 +119,7 @@ def encoder_forward(
output_hidden_states: Optional[bool] = False, output_hidden_states: Optional[bool] = False,
return_dict: Optional[bool] = True, return_dict: Optional[bool] = True,
): ):
if attention_mask and not attention_mask.any(): if attention_mask is not None and not attention_mask.any():
attention_mask = None attention_mask = None
return BertEncoder.forward( return BertEncoder.forward(
self=self, self=self,