Brett Kuprel df9aa6f915 sort -> topk, prev_token_and_index -> prev_token, token_index

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