Fix torch model
Merged