Edit torch model