Brett Kuprel 661ec976ac simplified attention for torch model

2022-06-29 13:48:12 -04:00

..

dalle_bart_decoder_flax.py

refactored to load models once and run multiple times

2022-06-29 09:42:12 -04:00

dalle_bart_decoder_torch.py

simplified attention for torch model

2022-06-29 13:48:12 -04:00

dalle_bart_encoder_flax.py

refactored to load models once and run multiple times

2022-06-29 09:42:12 -04:00

dalle_bart_encoder_torch.py

simplified attention for torch model

2022-06-29 13:48:12 -04:00

vqgan_detokenizer.py

works with cuda

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