"...composable_kernel_rocm.git" did not exist on "2559ef64c38fe06a52a8c8b84d672844119c4631"
Unverified Commit c40ecfd7 authored by Sylvain Gugger's avatar Sylvain Gugger Committed by GitHub
Browse files

Move import (#14787)

parent 7c9c41f4
......@@ -47,7 +47,6 @@ from .file_utils import (
is_remote_url,
)
from .generation_tf_utils import TFGenerationMixin
from .modelcard import TrainingSummary
from .modeling_tf_outputs import TFSeq2SeqLMOutput
from .tokenization_utils_base import BatchEncoding
from .utils import logging
......@@ -940,6 +939,9 @@ class TFPreTrainedModel(tf.keras.Model, TFModelUtilsMixin, TFGenerationMixin, Pu
dataset: Optional[Union[str, List[str]]] = None,
dataset_args: Optional[Union[str, List[str]]] = None,
):
# Avoids a circular import by doing this when necessary.
from .modelcard import TrainingSummary
training_summary = TrainingSummary.from_keras(
self,
keras_history=self.history,
......
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