From 534e143b0c927c1428d1a6ca0805fbda719fc4f2 Mon Sep 17 00:00:00 2001 From: Bryson Jones Date: Mon, 15 Dec 2025 10:08:47 -0800 Subject: [PATCH] add conditional transformers import to match all other policies that use transformers lib --- .../policies/multi_task_dit/modeling_multi_task_dit.py | 10 +++++++++- 1 file changed, 9 insertions(+), 1 deletion(-) diff --git a/src/lerobot/policies/multi_task_dit/modeling_multi_task_dit.py b/src/lerobot/policies/multi_task_dit/modeling_multi_task_dit.py index e646fd6fb..0d252b296 100644 --- a/src/lerobot/policies/multi_task_dit/modeling_multi_task_dit.py +++ b/src/lerobot/policies/multi_task_dit/modeling_multi_task_dit.py @@ -27,6 +27,7 @@ References: import math from collections import deque +from typing import TYPE_CHECKING import einops import torch @@ -36,9 +37,16 @@ import torchvision from diffusers.schedulers.scheduling_ddim import DDIMScheduler from diffusers.schedulers.scheduling_ddpm import DDPMScheduler from torch import Tensor -from transformers import CLIPTextModel, CLIPVisionModel from lerobot.policies.multi_task_dit.configuration_multi_task_dit import MultiTaskDiTConfig +from lerobot.utils.import_utils import _transformers_available + +# Conditional import for type checking and lazy loading +if TYPE_CHECKING or _transformers_available: + from transformers import CLIPTextModel, CLIPVisionModel +else: + CLIPTextModel = None + CLIPVisionModel = None from lerobot.policies.pretrained import PreTrainedPolicy from lerobot.policies.utils import populate_queues from lerobot.utils.constants import (