forked from umbertogriffo/rag-chatbot
-
Notifications
You must be signed in to change notification settings - Fork 0
/
Makefile
46 lines (37 loc) · 1.28 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
43
44
45
46
.PHONY: clean test
check:
which pip3
which python3
install:
echo "Installing..."
mkdir -p .venv
poetry config virtualenvs.in-project true
poetry install --no-root --no-ansi
echo "Installing torch and sentence-transformers with pip to avoid poetry's issues in installing torch... (it doesn't install CUDA dependencies)"
. .venv/bin/activate && pip3 install torch~=2.1.2 torchvision torchaudio
. .venv/bin/activate && pip3 install sentence-transformers~=2.2.2
echo "Installing llama-cpp-python with pip to get NVIDIA CUDA acceleration"
. .venv/bin/activate && CMAKE_ARGS="-DLLAMA_CUBLAS=on" pip3 install llama-cpp-python==0.2.28
install_pre_commit:
poetry run pre-commit install
poetry run pre-commit install --hook-type pre-commit
setup: install install_pre_commit
update:
poetry lock --no-update
poetry install
tidy:
poetry run ruff format --exclude=.venv .
poetry run ruff check --exclude=.venv . --fix
test:
poetry run pytest --log-cli-level=DEBUG --capture=tee-sys -v
check-formatting:
poetry run ruff format . --check
clean:
echo "Cleaning Poetry environment..."
rm -rf .venv
rm poetry.lock
echo "Poetry environment cleaned."
## Delete all compiled Python files
find . -type f -name "*.py[co]" -delete
find . -type d -name "__pycache__" -delete
rm -rf .pytest_cache