Fix flax import in __init__.py: modeling_xglm -> modeling_flax_xglm (#16556)

Fix flax import in init.py: modeling_xglm -> modeling_flax_xglm (#16556)
ec4da72f · Daniel Stancl · GitHub · 013a7dbe · ec4da72f
Unverified Commit ec4da72f authored Apr 04, 2022 by Daniel Stancl Committed by GitHub Apr 04, 2022
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

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

No files found.
--- a/src/transformers/models/xglm/__init__.py
+++ b/src/transformers/models/xglm/__init__.py
@@ -67,7 +67,7 @@ if TYPE_CHECKING:
        from .modeling_xglm import XGLM_PRETRAINED_MODEL_ARCHIVE_LIST, XGLMForCausalLM, XGLMModel, XGLMPreTrainedModel

    if is_flax_available():
-        from .modeling_xglm import FlaxXGLMForCausalLM, FlaxXGLMModel, FlaxXGLMPreTrainedModel
+        from .modeling_flax_xglm import FlaxXGLMForCausalLM, FlaxXGLMModel, FlaxXGLMPreTrainedModel


 else: