alanakbik's picture
Update model for torch 2.0
4e3f3d1