Skip to content

Commit 23a3b54

Browse files
committedFeb 6, 2025
fix format issue
Signed-off-by: Liu, Kaixuan <kaixuan.liu@intel.com>
1 parent 8dacb0a commit 23a3b54

File tree

1 file changed

+1
-3
lines changed

1 file changed

+1
-3
lines changed
 

‎optimum/exporters/ipex/modeling_utils.py

+1-3
Original file line numberDiff line numberDiff line change
@@ -312,9 +312,7 @@ def _falcon_model_forward(
312312
device = input_ids.device if input_ids is not None else inputs_embeds.device
313313

314314
if cache_position is None:
315-
cache_position = torch.arange(
316-
past_key_values_length, past_key_values_length + seq_length, device=device
317-
)
315+
cache_position = torch.arange(past_key_values_length, past_key_values_length + seq_length, device=device)
318316

319317
if position_ids is None:
320318
position_ids = cache_position.unsqueeze(0).repeat_interleave(input_ids.shape[0], 0)

0 commit comments

Comments
 (0)