Brett Kuprel 5aa6fe49bf use cuda if available

2022-06-28 12:47:11 -04:00

..

dalle_bart_decoder_flax.py

previous commit broke colab example, so adjusting flax requirement to 0.4.2 for now

2022-06-28 08:04:08 -04:00

dalle_bart_decoder_torch.py

use cuda if available

2022-06-28 12:47:11 -04:00

dalle_bart_encoder_flax.py

license and cleanup

2022-06-27 14:34:10 -04:00

dalle_bart_encoder_torch.py

torch.no_grad(), cleanup

2022-06-28 12:16:44 -04:00

vqgan_detokenizer.py

torch.no_grad(), cleanup

2022-06-28 12:16:44 -04:00