mirror of
https://github.com/huggingface/text-generation-inference.git
synced 2025-09-10 11:54:52 +00:00
First upgrade pip then run pip
Currently it will execute gen-server and install-transformers first, and then start upgrading pip. It should first upgrade pip.
This commit is contained in:
parent
70f485bf9f
commit
1175d0e4b4
@ -3,7 +3,7 @@ include Makefile-flash-att
|
||||
unit-tests:
|
||||
pytest -s -vv -m "not private" tests
|
||||
|
||||
gen-server:
|
||||
gen-server: upgrade-pip
|
||||
# Compile protos
|
||||
pip install grpcio-tools==1.51.1 mypy-protobuf==3.4.0 'types-protobuf>=3.20.4' --no-cache-dir
|
||||
mkdir text_generation_server/pb || true
|
||||
@ -12,12 +12,14 @@ gen-server:
|
||||
find text_generation_server/pb/ -type f -name "*.py" -print0 -exec sed -i -e 's/^\(import.*pb2\)/from . \1/g' {} \;
|
||||
touch text_generation_server/pb/__init__.py
|
||||
|
||||
install-torch:
|
||||
install-torch: upgrade-pip
|
||||
# Install specific version of torch
|
||||
pip install torch --extra-index-url https://download.pytorch.org/whl/cu118 --no-cache-dir
|
||||
|
||||
install: gen-server install-torch
|
||||
upgrade-pip:
|
||||
pip install pip --upgrade
|
||||
|
||||
install: gen-server install-torch upgrade-pip
|
||||
pip install -r requirements.txt
|
||||
pip install -e ".[bnb, accelerate]"
|
||||
|
||||
|
Loading…
Reference in New Issue
Block a user