gpt-pretrain/custom_models/gpt2/model.safetensors