File size: 751 Bytes
5f923cd
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
load("@rules_rust//rust:defs.bzl", "rust_library")

package(
    default_visibility = ["//visibility:public"],
)

rust_library(
    name = "huggingface_tokenizer_impl",
    srcs = ["rust/src/lib.rs"],
    edition = "2018",
    proc_macro_deps = [],
    deps = [
        "@crate_index//:serde_json",
        "@crate_index//:tokenizers",
    ],
)

cc_library(
    name = "huggingface_tokenizer",
    srcs = ["src/huggingface_tokenizer.cc"],
    hdrs = [
        "include/tokenizers_c.h",
        "include/tokenizers_cpp.h",
    ],
    includes = ["include"],
    deps = [
        ":huggingface_tokenizer_impl",
    ],
    linkopts = select({
        "@platforms//os:windows": ["kernel32.lib", "ntdll.lib"],
        "//conditions:default": [],
    }),
)