alanakbik's picture
Update model for torch 2.0
27fb2ba