Brett Kuprel 1e18ba0ffa is_expendable argument reduces memory usage for command line script

2022-06-30 06:43:10 -04:00

..

models

simplified flax attention and matched torch attention

2022-06-29 14:56:28 -04:00

load_params.py

works with cuda

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

min_dalle_base.py

is_expendable argument reduces memory usage for command line script

2022-06-30 06:43:10 -04:00

min_dalle_flax.py

is_expendable argument reduces memory usage for command line script

2022-06-30 06:43:10 -04:00

min_dalle_torch.py

is_expendable argument reduces memory usage for command line script

2022-06-30 06:43:10 -04:00

text_tokenizer.py

refactored to load models once and run multiple times

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