frozenleaves/pytorch
1
0
Fork 0
mirror of https://github.com/pytorch/pytorch.git synced 2025-10-20 21:14:14 +08:00
Code Issues Packages Projects Releases Wiki Activity

[functorch] Quick temporary fix for python key

Browse Source
This commit is contained in:
Richard Zou
2021-04-29 11:56:21 -07:00
committed by Jon Janzen
parent c521cd1f45
commit 2467773967
1 changed files with 6 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
functorch/functorch/__init__.py
View File

@ -4,7 +4,12 @@ from . import _C
from ._src.vmap import vmap from ._src.vmap import vmap
from ._src.eager_transforms import grad, grad_with_value, vjp, jacrev from ._src.eager_transforms import grad, grad_with_value, vjp, jacrev
from ._src.make_functional import make_functional, make_functional_with_buffers from ._src.make_functional import make_functional, make_functional_with_buffers
from ._src.python_key import key_wrap, ModuleWrap
# Python key stuff is not in pytorch/pytorch core
try:
from ._src.python_key import key_wrap, ModuleWrap
except:
pass
# Monkeypatching lol # Monkeypatching lol