Brett Kuprel b40fd83a0d mega works with latest flax version 0.5.2 now, removing 0.4.2 pin		2022-07-01 02:58:43 -04:00
..
dalle_bart_decoder_flax.py	mega works with latest flax version 0.5.2 now, removing 0.4.2 pin	2022-07-01 02:58:43 -04:00
dalle_bart_decoder_torch.py	sort -> topk, prev_token_and_index -> prev_token, token_index	2022-06-30 09:04:11 -04:00
dalle_bart_encoder_flax.py	delete cache	2022-06-30 15:48:20 -04:00
dalle_bart_encoder_torch.py	simplified attention for torch model	2022-06-29 13:48:12 -04:00
vqgan_detokenizer.py	sort -> topk, prev_token_and_index -> prev_token, token_index	2022-06-30 09:04:11 -04:00