Skip to content

Commit aec45a9

Browse files
committed
fix type declaration
1 parent 0ecaa0f commit aec45a9

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

optimum/intel/openvino/modeling_decoder.py

+3-3
Original file line numberDiff line numberDiff line change
@@ -17,7 +17,7 @@
1717
from dataclasses import dataclass
1818
from pathlib import Path
1919
from tempfile import TemporaryDirectory
20-
from typing import Dict, Optional, Tuple, Union
20+
from typing import Any, Dict, Optional, Tuple, Union
2121

2222
import numpy as np
2323
import openvino
@@ -504,10 +504,10 @@ def forward(
504504
def _update_model_kwargs_for_generation(
505505
self,
506506
outputs: OVCausalLMOutputWithPast,
507-
model_kwargs: Dict[str],
507+
model_kwargs: Dict[str, Any],
508508
is_encoder_decoder: bool = False,
509509
standardize_cache_format: bool = False,
510-
) -> Dict[str]:
510+
) -> Dict[str, Any]:
511511
model_kwargs = super()._update_model_kwargs_for_generation(
512512
outputs=outputs,
513513
model_kwargs=model_kwargs,

0 commit comments

Comments
 (0)