-
Notifications
You must be signed in to change notification settings - Fork 4
/
Copy pathMakefile
42 lines (34 loc) · 1.57 KB
/
Makefile
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
35
36
37
38
39
40
41
42
targets = \
setup \
output/tinyllama-color-coder-v1/checkpoint-200/README.md \
output/tinyllama-sql-coder-v1/checkpoint-200/README.md
all: $(targets)
.PHONY: setup lock
setup:
conda install -c nvidia cuda-toolkit=12.1
conda install pytorch torchvision torchaudio pytorch-cuda=12.1 -c pytorch -c nvidia
pip install -U pip
pip install packaging
pip install flash-attn --no-build-isolation
pip install -Ur requirements.txt
lock:
pip freeze > requirements.txt.lock
.PHONY: eval-all
eval-all: $(targets)
python3 src/eval.py recipes/RTX_3060_12GB/color-coder.yaml
python3 src/eval.py recipes/RTX_3060_12GB/sql-coder.yaml
output/tinyllama-color-coder-v1/checkpoint-200/README.md:
accelerate launch src/train.py recipes/RTX_3060_12GB/color-coder.yaml
output/tinyllama-sql-coder-v1/checkpoint-200/README.md:
accelerate launch src/train.py recipes/RTX_3060_12GB/sql-coder.yaml
docker:
docker build --no-cache -t yuiseki/infinite-tinyllama:latest .
docker-run:
docker run -it --rm --gpus all -v $(PWD):/app yuiseki/infinite-tinyllama:latest
ollama:
ollama create elyza-llama2:7b-instruct -f ollama_models/ELYZA/Llama-2/7b-instruct/Modelfile
ollama create elyza-codellama:7b-instruct -f ollama_models/ELYZA/CodeLlama/7b-instruct/Modelfile
ollama create rakutenai:7b-instruct -f ollama_models/Rakuten/RakutenAI/7b-instruct/Modelfile
ollama create rakutenai:7b-chat -f ollama_models/Rakuten/RakutenAI/7b-chat/Modelfile
ollama create rinna-youri:7b-instruct -f ollama_models/rinna/youri/7b-instruct/Modelfile
ollama create rinna-youri:7b-chat -f ollama_models/rinna/youri/7b-chat/Modelfile