Fix Torch for Omni and change repo.