alanakbik's picture
Update model for torch 2.0
34cdda2