Fix import in function

huggingface · May 13, 2022 · 373feb6 · 373feb6
1 parent b69ef39
commit 373feb6
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/src/transformers/utils/fx.py b/src/transformers/utils/fx.py
@@ -26,6 +26,7 @@
 from packaging import version
 from torch import nn
 from torch.fx import Graph, GraphModule, Proxy, Tracer
+from torch.fx.proxy import ParameterProxy
 
 from .. import (
     CONFIG_MAPPING,
@@ -638,8 +639,6 @@ def create_proxy(self, kind, target, args, kwargs, name=None, type_expr=None, pr
         return rv
 
     def _module_getattr(self, attr, attr_val, parameter_proxy_cache):
-        from torch.fx.proxy import ParameterProxy
-
         if getattr(self, "_disable_module_getattr", False):
             return attr_val
         else: