From b5e27496fb5970e513391775dcebc56a131fb2e1 Mon Sep 17 00:00:00 2001 From: Prisha Jain Date: Thu, 23 Jul 2026 11:26:46 +0530 Subject: [PATCH] Fixing nightly regression test --- src/maxdiffusion/max_utils.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/src/maxdiffusion/max_utils.py b/src/maxdiffusion/max_utils.py index 1ab5407ac..e2f8cca04 100644 --- a/src/maxdiffusion/max_utils.py +++ b/src/maxdiffusion/max_utils.py @@ -55,7 +55,11 @@ from flax.training import train_state from jax.experimental import mesh_utils -from transformers import FlaxCLIPTextModel, FlaxCLIPTextPreTrainedModel +try: + from transformers import FlaxCLIPTextModel, FlaxCLIPTextPreTrainedModel +except ImportError: + FlaxCLIPTextModel = None + FlaxCLIPTextPreTrainedModel = None from flax import struct from flax import core from jax.experimental.pallas.ops.tpu.splash_attention import splash_attention_kernel