Skip to content

Commit 7a929e8

Browse files
authored
Fix backward compatibility for GenerationMode import (#719)
1 parent 7114900 commit 7a929e8

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

optimum/intel/openvino/modeling_decoder.py

+2-2
Original file line numberDiff line numberDiff line change
@@ -28,10 +28,10 @@
2828
from transformers import AutoModelForCausalLM, PretrainedConfig
2929
from transformers.file_utils import add_start_docstrings, add_start_docstrings_to_model_forward
3030
from transformers.generation import GenerationMixin
31-
from transformers.generation.configuration_utils import GenerationConfig, GenerationMode
31+
from transformers.generation.configuration_utils import GenerationConfig
3232
from transformers.generation.logits_process import LogitsProcessorList
3333
from transformers.generation.stopping_criteria import StoppingCriteriaList
34-
from transformers.generation.utils import GenerateOutput
34+
from transformers.generation.utils import GenerateOutput, GenerationMode
3535
from transformers.modeling_outputs import CausalLMOutputWithPast
3636

3737
from optimum.utils.normalized_config import NormalizedConfigManager

0 commit comments

Comments
 (0)