File tree 5 files changed +6
-88
lines changed
pytorch_tokenizers/tools/llama2c
5 files changed +6
-88
lines changed Original file line number Diff line number Diff line change 11
11
12
12
import argparse
13
13
14
- from pytorch_tokenizers import Llama2cTokenizer
14
+ from pytorch_tokenizers . llama2c import Llama2cTokenizer
15
15
16
16
17
17
if __name__ == "__main__" :
Original file line number Diff line number Diff line change @@ -24,6 +24,9 @@ def define_common_targets():
24
24
external_deps = [
25
25
"sentencepiece-py" ,
26
26
],
27
+ deps = [
28
+ "//pytorch/tokenizers/pytorch_tokenizers:tokenizers" ,
29
+ ],
27
30
)
28
31
29
32
runtime .python_binary (
Original file line number Diff line number Diff line change 5
5
# LICENSE file in the root directory of this source tree.
6
6
# @lint-ignore-every LICENSELINT
7
7
# type: ignore[syntax]
8
- from setuptools import find_packages , setup
8
+ from setuptools import setup
9
9
10
10
with open ("README.md" , "r" ) as f :
11
11
long_description = f .read ()
14
14
version = "0.1.0" ,
15
15
long_description = long_description ,
16
16
long_description_content_type = "text/markdown" ,
17
- packages = find_packages (where = "pytorch_tokenizers" ),
18
- package_dir = {"" : "pytorch_tokenizers" },
17
+ packages = ["pytorch_tokenizers" ],
19
18
)
Load Diff This file was deleted.
Load Diff This file was deleted.
You can’t perform that action at this time.
0 commit comments