Makefile-transformers 501 Bytes
Newer Older
1
transformers_commit := 69009822aa7897ffab97afb814e38126b83f639e
2

3
4
transformers:
	# Clone fork of transformers with custom CUDA kernels and sharding logic
5
6
	pip install --upgrade setuptools
	git clone https://github.com/OlivierDehaene/transformers.git
7
8
9
10
11
12

build-transformers: transformers
	cd transformers && git fetch && git checkout $(transformers_commit) && python setup.py build

install-transformers: build-transformers
	pip uninstall transformers -y || true
13
	cd transformers && python setup.py install