More fix and update OmniTorch