AIForce/train-dalle-together

Bez popisu

7 Větve

Alexander Borzunov b67665d0eb Add LICENSE		před 2 roky
inference	acf688eac2 Use --top_p and --top_k options in run_inference.py	před 3 roky
lib	64dee420da Upgrade to using hivemind.optim.experimental	před 4 roky
.gitignore	72fc0bcdb7 Initial commit (ru-max branch without private code)	před 4 roky
LICENSE	b67665d0eb Add LICENSE	před 2 roky
README.md	d4140807f4 Update readme	před 4 roky
arguments.py	19e3d2d060 Make aux peers fetch checkpoints every 2 steps by default	před 4 roky
callback.py	4918c58cb6 Polish stdout	před 4 roky
data.py	c61c61b20d Use t5-small tokenizer	před 4 roky
huggingface_auth.py	3e604bc1f5 fix auth	před 4 roky
manage_scaleset.py	c365b2ec9f Tweak settings for the upcoming demo (#2)	před 4 roky
requirements.txt	872db56e0b Fix bitsandbytes version	před 3 roky
run_aux_peer.py	748f20991c Make runners executable and improve their shebangs	před 4 roky
run_trainer.py	748f20991c Make runners executable and improve their shebangs	před 4 roky
run_trainer_tpu.py	748f20991c Make runners executable and improve their shebangs	před 4 roky
task.py	09240991cc Make model uploading use access token from authorizer (#7)	před 4 roky
utils.py	f621362466 Make logging less verbose	před 4 roky

Training DALL-E with volunteers from all over the Internet

This repository is a part of the NeurIPS 2021 demonstration "Training Transformers Together".

In this demo, we train a model similar to OpenAI DALL-E — a Transformer "language model" that generates images from text descriptions. Training happens collaboratively — volunteers from all over the Internet contribute to the training using hardware available to them. We use LAION-400M, the world's largest openly available image-text-pair dataset with 400 million samples. Our model is based on the dalle‑pytorch implementation by Phil Wang with a few tweaks to make it communication-efficient.

See details about how to join and how it works on our website.

README.md

Training DALL-E with volunteers from all over the Internet