alanakbik's picture
Update model for torch 2.0
9e5dc17