Unverified Commit c4d4e8bd authored by Yacine Jernite's avatar Yacine Jernite Committed by GitHub
Browse files

Move GenerationMixin to separate file (#5254)

* separate_generation_code

* isort

* renamed

* rename_files

* move_shapelit
parent 90d13954
...@@ -169,7 +169,8 @@ if is_sklearn_available(): ...@@ -169,7 +169,8 @@ if is_sklearn_available():
# Modeling # Modeling
if is_torch_available(): if is_torch_available():
from .modeling_utils import PreTrainedModel, prune_layer, Conv1D, top_k_top_p_filtering, apply_chunking_to_forward from .generation_utils import top_k_top_p_filtering
from .modeling_utils import PreTrainedModel, prune_layer, Conv1D, apply_chunking_to_forward
from .modeling_auto import ( from .modeling_auto import (
AutoModel, AutoModel,
AutoModelForPreTraining, AutoModelForPreTraining,
...@@ -406,9 +407,9 @@ if is_torch_available(): ...@@ -406,9 +407,9 @@ if is_torch_available():
# TensorFlow # TensorFlow
if is_tf_available(): if is_tf_available():
from .generation_tf_utils import tf_top_k_top_p_filtering
from .modeling_tf_utils import ( from .modeling_tf_utils import (
shape_list, shape_list,
tf_top_k_top_p_filtering,
TFPreTrainedModel, TFPreTrainedModel,
TFSequenceSummary, TFSequenceSummary,
TFSharedEmbeddings, TFSharedEmbeddings,
......
This diff is collapsed.
This diff is collapsed.
This diff is collapsed.
This diff is collapsed.
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