added OmniTorch support