Skip to content

Commit 018d632

Browse files
committed
Update _utils.py
1 parent 647f894 commit 018d632

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

Diff for: unsloth/models/_utils.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -107,7 +107,7 @@ def _get_pad_token_id(self, pad_token_id: int = None, eos_token_id: int = None)
107107
# logger.warning(f"Setting `pad_token_id` to `eos_token_id`:{eos_token_id} for open-end generation.")
108108
pad_token_id = eos_token_id
109109
return pad_token_id
110-
from transformers.generation_utils import GenerationMixin
110+
from transformers import GenerationMixin
111111
GenerationMixin._get_pad_token_id = _get_pad_token_id
112112

113113

0 commit comments

Comments
 (0)