cuda_2d_sr

arguments.py

data_utils/datasets.py

data_utils/vqvae_tokenizer.py

draw_diff.py

generate_samples.py

generation/__init__.py

generation/cuda_2d_sampling.py

generation/sampling.py

learning_rates.py

model/gpt2_modeling.py

mpu/sparse_transformer.py

mpu/unused_codes.py

pretrain_gpt2.py

random_display.py

scripts/cuda_2d_text2image.sh

scripts/pretrain_multiple_nodes.sh

scripts/testnan.sh

scripts/text2image.sh

test_sparse_attention.py

utils.py