From 3236f798efae575a4e468db3c04e8d787df8d0f0 Mon Sep 17 00:00:00 2001 From: id01 Date: Mon, 25 Aug 2025 11:16:23 -0700 Subject: [PATCH 1/2] Fix https://github.com/huggingface/transformers/issues/40292 --- src/transformers/integrations/integration_utils.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/src/transformers/integrations/integration_utils.py b/src/transformers/integrations/integration_utils.py index 702e05627cb6..fc3b986aedd1 100755 --- a/src/transformers/integrations/integration_utils.py +++ b/src/transformers/integrations/integration_utils.py @@ -41,7 +41,7 @@ if os.getenv("WANDB_MODE") == "offline": print("⚙️ Running in WANDB offline mode") -from .. import PreTrainedModel, TFPreTrainedModel, TrainingArguments +from .. import PreTrainedModel, TrainingArguments from .. import __version__ as version from ..utils import ( PushToHubMixin, @@ -56,6 +56,9 @@ logger = logging.get_logger(__name__) +if is_tf_available(): + from .. import TFPreTrainedModel + if is_torch_available(): import torch import torch.distributed as dist From f701e54125e1154d97f456a23f1f7ad126808af8 Mon Sep 17 00:00:00 2001 From: Matt Date: Mon, 25 Aug 2025 19:51:18 +0100 Subject: [PATCH 2/2] Trigger tests