Revert "fix omni torch"

This reverts commit 7f436b6c853125122fe7692bdf9deee913be3674.
Revert "lets try omni torch"

This reverts commit 24a4c6dc7342afda00750ec00befbe4654a7aabc.