File tree Expand file tree Collapse file tree 1 file changed +2
-2
lines changed Expand file tree Collapse file tree 1 file changed +2
-2
lines changed Original file line number Diff line number Diff line change 2828from transformers import AutoModelForCausalLM , PretrainedConfig
2929from transformers .file_utils import add_start_docstrings , add_start_docstrings_to_model_forward
3030from transformers .generation import GenerationMixin
31- from transformers .generation .configuration_utils import GenerationConfig , GenerationMode
31+ from transformers .generation .configuration_utils import GenerationConfig
3232from transformers .generation .logits_process import LogitsProcessorList
3333from transformers .generation .stopping_criteria import StoppingCriteriaList
34- from transformers .generation .utils import GenerateOutput
34+ from transformers .generation .utils import GenerateOutput , GenerationMode
3535from transformers .modeling_outputs import CausalLMOutputWithPast
3636
3737from optimum .utils .normalized_config import NormalizedConfigManager
You can’t perform that action at this time.
0 commit comments