Brett Kuprel d99828a239 simplified flax attention and matched torch attention		2022-06-29 14:56:28 -04:00
..
dalle_bart_decoder_flax.py	simplified flax attention and matched torch attention	2022-06-29 14:56:28 -04:00
dalle_bart_decoder_torch.py	simplified flax attention and matched torch attention	2022-06-29 14:56:28 -04:00
dalle_bart_encoder_flax.py	simplified flax attention and matched torch attention	2022-06-29 14:56:28 -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