File tree 1 file changed +2
-2
lines changed
1 file changed +2
-2
lines changed Original file line number Diff line number Diff line change 28
28
from transformers import AutoModelForCausalLM , PretrainedConfig
29
29
from transformers .file_utils import add_start_docstrings , add_start_docstrings_to_model_forward
30
30
from transformers .generation import GenerationMixin
31
- from transformers .generation .configuration_utils import GenerationConfig , GenerationMode
31
+ from transformers .generation .configuration_utils import GenerationConfig
32
32
from transformers .generation .logits_process import LogitsProcessorList
33
33
from transformers .generation .stopping_criteria import StoppingCriteriaList
34
- from transformers .generation .utils import GenerateOutput
34
+ from transformers .generation .utils import GenerateOutput , GenerationMode
35
35
from transformers .modeling_outputs import CausalLMOutputWithPast
36
36
37
37
from optimum .utils .normalized_config import NormalizedConfigManager
You can’t perform that action at this time.
0 commit comments