Closed weifengpy closed 3 months ago
train.py renamed model to whole_model https://github.com/pytorch/torchtitan/pull/406
model
whole_model
fp8 still use model thus report error on model not defined. this PR fixed it
model not defined
build_fp8_linear(whole_model, job_config)
train.py renamed
modeltowhole_modelhttps://github.com/pytorch/torchtitan/pull/406fp8 still use
modelthus report error onmodel not defined. this PR fixed itbuild_fp8_linear(whole_model, job_config)