diff --git a/mindtorch/__init__.py b/mindtorch/__init__.py index dfe5a798e..58aa0f478 100644 --- a/mindtorch/__init__.py +++ b/mindtorch/__init__.py @@ -172,7 +172,7 @@ def _running_with_deploy(): from .storage import UntypedStorage, Storage, TypedStorage from . import _dynamo, library -from . import profiler, cuda, npu, amp, compiler, jit, version, __future__, overrides, \ +from . import profiler, cuda, npu, xpu, mps, amp, compiler, jit, version, __future__, overrides, \ return_types, linalg, fx, backends, nn, fft, _jit_internal, utils, optim, testing, _ops from ._lowrank import svd_lowrank from .random import get_rng_state, initial_seed, manual_seed, seed, set_rng_state @@ -182,4 +182,3 @@ def _running_with_deploy(): from .torch_proxy import initialize_torch_proxy, setup_metadata_patch initialize_torch_proxy() setup_metadata_patch() - diff --git a/requirements/requirements.txt b/requirements/requirements.txt index fb07db3bb..16868c8a3 100644 --- a/requirements/requirements.txt +++ b/requirements/requirements.txt @@ -37,4 +37,6 @@ transformers peft diffusers timeout_decorator -torch \ No newline at end of file +torch +ftfy +requests_mock