revert requirements

Author: eltsai

dependencies/requirements/generated_requirements/requirements.txt

@@ -2,7 +2,6 @@
 # If you need to modify dependencies, please do so in the host requirements file and run seed-env again.
 
 absl-py>=2.3.1
-accelerate>=1.13.0
 aiofiles>=25.1.0
 aiohappyeyeballs>=2.6.1
 aiohttp>=3.13.3
@@ -68,7 +67,7 @@ hf-transfer>=0.1.9
 hf-xet>=1.4.2 ; platform_machine == 'aarch64' or platform_machine == 'amd64' or platform_machine == 'arm64' or platform_machine == 'x86_64'
 httpcore>=1.0.9
 httpx>=0.28.1
-huggingface-hub>=1.10.1
+huggingface-hub>=0.36.2
 humanize>=4.15.0
 hypothesis>=6.142.1
 idna>=3.11
@@ -81,7 +80,6 @@ isort>=8.0.1
 jaraco-functools>=4.4.0
 jax>=0.9.0
 jaxlib>=0.9.0
-jaxopt>=0.8.5
 jaxtyping>=0.3.9
 jinja2>=3.1.6
 keras>=3.13.1

src/maxdiffusion/generate_flux.py

@@ -29,19 +29,7 @@
 from chex import Array
 from einops import rearrange
 from flax.linen import partitioning as nn_partitioning
-
-try:
-  from transformers import (CLIPTokenizer, FlaxCLIPTextModel, T5EncoderModel, FlaxT5EncoderModel, AutoTokenizer)
-except ImportError:
-  # For transformers>=5.0, Flax models have different import paths
-  from transformers import CLIPTokenizer, T5EncoderModel, AutoTokenizer
-
-  try:
-    from transformers.models.clip.modeling_flax_clip import FlaxCLIPTextModel
-    from transformers.models.t5.modeling_flax_t5 import FlaxT5EncoderModel
-  except ImportError:
-    FlaxCLIPTextModel = None
-    FlaxT5EncoderModel = None
+from transformers import (CLIPTokenizer, FlaxCLIPTextModel, T5EncoderModel, FlaxT5EncoderModel, AutoTokenizer)
 
 from maxdiffusion import FlaxAutoencoderKL, pyconfig, max_logging
 from maxdiffusion.models.flux.transformers.transformer_flux_flax import FluxTransformer2DModel