mirror of
https://github.com/huggingface/transformers.git
synced 2025-07-03 21:00:08 +06:00
nit
This commit is contained in:
parent
613a0f7bdf
commit
ca5a0a2b56
@ -314,6 +314,7 @@ class TorchExportableModuleWithStaticCache(torch.nn.Module):
|
||||
kwargs = {
|
||||
"input_ids": input_ids,
|
||||
"position_ids": position_ids,
|
||||
"attention_mask": None,
|
||||
"cache_position": cache_position,
|
||||
"past_key_values": past_key_values,
|
||||
"use_cache": True,
|
||||
|
Loading…
Reference in New Issue
Block a user