keep transformers < 5.0.0 for now

Author: eltsai

src/maxdiffusion/max_utils.py

@@ -47,15 +47,7 @@
 from flax.training import train_state
 from jax.experimental import mesh_utils
 
-try:
-  from transformers import (FlaxCLIPTextModel, FlaxCLIPTextPreTrainedModel)
-except ImportError:
-  # For transformers>=5.0, these need different import paths
-  try:
-    from transformers.models.clip.modeling_flax_clip import FlaxCLIPTextModel, FlaxCLIPTextPreTrainedModel
-  except ImportError:
-    FlaxCLIPTextModel = None
-    FlaxCLIPTextPreTrainedModel = None
+from transformers import FlaxCLIPTextModel, FlaxCLIPTextPreTrainedModel
 from flax import struct
 from typing import (
     Callable,