test load model, perform normally

.gitignore

arguments.py

data_utils/configure_data.py

model/__init__.py

model/base_model.py

model/cuda2d_model.py

move_weights.py

mpu/__init__.py

mpu/layers.py

mpu/random.py

mpu/transformer.py

pretrain_cogview2.py

pretrain_gpt2.py

scripts/ds_config_zero.json

scripts/pretrain_cogview2.sh

scripts/pretrain_multiple_nodes.sh

scripts_old/cuda_2d_text2image.sh

scripts_old/image2text.sh

scripts_old/low_level_super_resolution.sh

scripts_old/post_selection.sh