ullm/WORKSPACE at main · aarossig/ullm · GitHub

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("@bazel_tools//tools/build_defs/repo:git.bzl", "git_repository")
load("@bazel_tools//tools/build_defs/repo:http.bzl", "http_file")

# c-flags ######################################################################

git_repository(
    name = "cflags",
    commit = "e73b8666ce88f63e7944244b94b0ed1c0488dab0",
    remote = "https://github.com/DieTime/c-flags.git",
    build_file = "@//build:cflags.bzl",
)

# llama2 #######################################################################

git_repository(
    name = "llama2",
    commit = "350e04fe35433e6d2941dce5a1f53308f87058eb",
    remote = "https://github.com/karpathy/llama2.c.git",
    build_file = "@//build:llama2.bzl",
)

http_file(
    name = "tinystories15M",
    url = "https://huggingface.co/karpathy/tinyllamas/resolve/main/stories15M.bin",
    integrity = "sha256-zVkGRNljhnorbloRB/UfrWY8QdecFJ++y7sflfqB9Jo=",
    downloaded_file_path = "tinystories15M.bin",
)

http_file(
    name = "tinystories110M",
    url = "https://huggingface.co/karpathy/tinyllamas/resolve/main/stories110M.bin",
    integrity = "sha256-UVJnFocmoe0TF6ZKQISS5q87Z8H3HFvZjAHZ1yGAOiQ=",
    downloaded_file_path = "tinystories110M.bin",
)