tmp save

draw_diff.py

finetune/__init__.py

fp16/__init__.py

fp16/fp16.py

fp16/fp16util.py

fp16/loss_scaler.py

generate_samples.py

model/__init__.py

model/base_model.py

model/distributed.py

model/gpt2.py

model/gpt2_modeling.py

model/mixins.py

mpu/__init__.py

mpu/grads.py

mpu/sparse_transformer.py

mpu/transformer.py

mpu/utils.py

pretrain_gpt2.py

utils.py