Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension


Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
3 changes: 1 addition & 2 deletions mindtorch/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand All @@ -182,4 +182,3 @@ def _running_with_deploy():
from .torch_proxy import initialize_torch_proxy, setup_metadata_patch
initialize_torch_proxy()
setup_metadata_patch()

4 changes: 3 additions & 1 deletion requirements/requirements.txt
Original file line number Diff line number Diff line change
Expand Up @@ -37,4 +37,6 @@ transformers
peft
diffusers
timeout_decorator
torch
torch
ftfy
requests_mock