Skip to content

GitLab

"git@developer.sourcefind.cn:chenpangpang/transformers.git" did not exist on "a3f96f366a49bbe2cbdeaebd2e32ccdc1260a1d6"
Unverified Commit 6b353264 authored by Sylvain Gugger's avatar Sylvain Gugger Committed by GitHub
Browse files

Properly register missing submodules in main init (#13372)

parent 4b7988eb
Hide whitespace changes
Inline Side-by-side
Showing with 4 additions and 0 deletions
src/transformers/__init__.py
View file @ 6b353264
...@@ -516,6 +516,7 @@ if is_torch_available(): ...@@ -516,6 +516,7 @@ if is_torch_available():
"StoppingCriteriaList", "StoppingCriteriaList",
] ]
_import_structure["generation_utils"] = ["top_k_top_p_filtering"] _import_structure["generation_utils"] = ["top_k_top_p_filtering"]
_import_structure["modeling_outputs"] = []
_import_structure["modeling_utils"] = ["Conv1D", "PreTrainedModel", "apply_chunking_to_forward", "prune_layer"] _import_structure["modeling_utils"] = ["Conv1D", "PreTrainedModel", "apply_chunking_to_forward", "prune_layer"]
# PyTorch models structure # PyTorch models structure
...@@ -1249,6 +1250,7 @@ if is_tf_available(): ...@@ -1249,6 +1250,7 @@ if is_tf_available():
_import_structure["benchmark.benchmark_args_tf"] = ["TensorFlowBenchmarkArguments"] _import_structure["benchmark.benchmark_args_tf"] = ["TensorFlowBenchmarkArguments"]
_import_structure["benchmark.benchmark_tf"] = ["TensorFlowBenchmark"] _import_structure["benchmark.benchmark_tf"] = ["TensorFlowBenchmark"]
_import_structure["generation_tf_utils"] = ["tf_top_k_top_p_filtering"] _import_structure["generation_tf_utils"] = ["tf_top_k_top_p_filtering"]
_import_structure["modeling_tf_outputs"] = []
_import_structure["modeling_tf_utils"] = [ _import_structure["modeling_tf_utils"] = [
"TFPreTrainedModel", "TFPreTrainedModel",
"TFSequenceSummary", "TFSequenceSummary",
...@@ -1675,6 +1677,8 @@ if is_flax_available(): ...@@ -1675,6 +1677,8 @@ if is_flax_available():
"FlaxTopKLogitsWarper", "FlaxTopKLogitsWarper",
"FlaxTopPLogitsWarper", "FlaxTopPLogitsWarper",
] ]
_import_structure["generation_flax_utils"] = []
_import_structure["modeling_flax_outputs"] = []
_import_structure["modeling_flax_utils"] = ["FlaxPreTrainedModel"] _import_structure["modeling_flax_utils"] = ["FlaxPreTrainedModel"]
_import_structure["models.albert"].extend( _import_structure["models.albert"].extend(
[ [
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment