gen-server:
	# Compile protos
	pip install grpcio-tools==1.49.1 --no-cache-dir
	mkdir text_generation/pb || true
	python -m grpc_tools.protoc -I../proto --python_out=text_generation/pb --grpc_python_out=text_generation/pb ../proto/generate.proto
	find text_generation/pb/ -type f -name "*.py" -print0 -exec sed -i -e 's/^\(import.*pb2\)/from . \1/g' {} \;
	touch text_generation/pb/__init__.py

install-transformers:
	# Install specific version of transformers
	rm transformers || true
	rm transformers-7302a24535e8dc5637ea5b4e4572fc971d404098 || true
	curl -L -O https://github.com/OlivierDehaene/transformers/archive/7302a24535e8dc5637ea5b4e4572fc971d404098.zip
	unzip 7302a24535e8dc5637ea5b4e4572fc971d404098.zip
	rm 7302a24535e8dc5637ea5b4e4572fc971d404098.zip
	mv transformers-7302a24535e8dc5637ea5b4e4572fc971d404098 transformers
	cd transformers && python setup.py install

install-safetensors:
	# Install specific version of safetensors
	pip install setuptools_rust
	rm safetensors || true
	rm safetensors-634deccbcbad5eaf417935281f8b3be7ebca69c5 || true
	curl -L -O https://github.com/huggingface/safetensors/archive/634deccbcbad5eaf417935281f8b3be7ebca69c5.zip
	unzip 634deccbcbad5eaf417935281f8b3be7ebca69c5.zip
	rm 634deccbcbad5eaf417935281f8b3be7ebca69c5.zip
	mv safetensors-634deccbcbad5eaf417935281f8b3be7ebca69c5 safetensors
	cd safetensors/bindings/python && python setup.py develop

install-torch:
	# Install specific version of torch
	pip install torch --extra-index-url https://download.pytorch.org/whl/cu116 --no-cache-dir

install: gen-server install-torch install-transformers install-safetensors
	pip install pip --upgrade
	pip install -e . --no-cache-dir

run-dev:
	python -m torch.distributed.run --nproc_per_node=2 text_generation/cli.py serve bigscience/bloom-560m --sharded