also remove torch from framework