Brett Kuprel 17c96fe110 works with cuda

2022-06-28 21:28:36 -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

works with cuda

2022-06-28 21:28:36 -04:00

dalle_bart_encoder_flax.py

license and cleanup

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

dalle_bart_encoder_torch.py

works with cuda

2022-06-28 21:28:36 -04:00

vqgan_detokenizer.py

works with cuda

2022-06-28 21:28:36 -04:00