min-dalle-test/min_dalle/models
2022-06-30 09:04:11 -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 sort -> topk, prev_token_and_index -> prev_token, token_index 2022-06-30 09:04:11 -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 sort -> topk, prev_token_and_index -> prev_token, token_index 2022-06-30 09:04:11 -04:00