Spaces:

exbert-project
/

exbert

Running on CPU Upgrade

git subrepo clone (merge) --branch=exbert-mods https://github.com/bhoov/transformers.git server/transformers

75466df about 5 years ago

186 Bytes

	FROM pytorch/pytorch:latest

	RUN git clone https://github.com/NVIDIA/apex.git && cd apex && python setup.py install --cuda_ext --cpp_ext

	RUN pip install transformers

	WORKDIR /workspace