Yucheng Tu

Thanks for checking in. I just tried to restrict the use to a single GPU, the original Faketensor issue is gone and training goes normal.