alanakbik's picture
Update model for torch 2.0
4f31790