Add module_spec to new model

126bddd1 · Sylvain Gugger · c962c2ad · 126bddd1
Commit 126bddd1 authored Jan 21, 2022 by Sylvain Gugger
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

src/transformers/models/swin/__init__.py src/transformers/models/swin/__init__.py +1 -1

No files found.
--- a/src/transformers/models/swin/__init__.py
+++ b/src/transformers/models/swin/__init__.py
@@ -50,4 +50,4 @@ if TYPE_CHECKING:
 else:
    import sys

-    sys.modules[__name__] = _LazyModule(__name__, globals()["__file__"], _import_structure)
+    sys.modules[__name__] = _LazyModule(__name__, globals()["__file__"], _import_structure, module_spec=__spec__)