Skip to content

Commit 1051e8c

Browse files
committed
Remove top-level imports of transformers and datasets
1 parent facac71 commit 1051e8c

File tree

1 file changed

+4
-2
lines changed

1 file changed

+4
-2
lines changed

outlines/models/transformers.py

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,8 +1,6 @@
11
from typing import TYPE_CHECKING, List, Optional, Tuple, Union
22

33
import torch
4-
from datasets.fingerprint import Hasher
5-
from transformers.file_utils import SPIECE_UNDERLINE
64

75
from outlines.models.tokenizer import Tokenizer
86

@@ -157,6 +155,8 @@ def decode(self, token_ids: torch.LongTensor) -> List[str]:
157155
return text
158156

159157
def convert_token_to_string(self, token: str) -> str:
158+
from transformers.file_utils import SPIECE_UNDERLINE
159+
160160
string = self.tokenizer.convert_tokens_to_string([token])
161161

162162
if self.is_llama:
@@ -172,6 +172,8 @@ def __eq__(self, other):
172172
return NotImplemented
173173

174174
def __hash__(self):
175+
from datasets.fingerprint import Hasher
176+
175177
return hash(Hasher.hash(self.tokenizer))
176178

177179

0 commit comments

Comments
 (0)