diff --git a/src/diffusers/models/attention_flax.py b/src/diffusers/models/attention_flax.py
index 1bde62e5c666..4d5471961f64 100644
--- a/src/diffusers/models/attention_flax.py
+++ b/src/diffusers/models/attention_flax.py
@@ -157,7 +157,7 @@ class FlaxAttention(nn.Module):
 
     def setup(self):
         logger.warning(
-            "Flax classes are deprecated and will be removed in Diffusers v1.0.0. We "
+            "Flax classes are deprecated and will be removed in Diffusers v0.40.0. We "
             "recommend migrating to PyTorch classes or pinning your version of Diffusers."
         )
 
@@ -288,7 +288,7 @@ class FlaxBasicTransformerBlock(nn.Module):
 
     def setup(self):
         logger.warning(
-            "Flax classes are deprecated and will be removed in Diffusers v1.0.0. We "
+            "Flax classes are deprecated and will be removed in Diffusers v0.40.0. We "
             "recommend migrating to PyTorch classes or pinning your version of Diffusers."
         )
 
@@ -381,7 +381,7 @@ class FlaxTransformer2DModel(nn.Module):
 
     def setup(self):
         logger.warning(
-            "Flax classes are deprecated and will be removed in Diffusers v1.0.0. We "
+            "Flax classes are deprecated and will be removed in Diffusers v0.40.0. We "
             "recommend migrating to PyTorch classes or pinning your version of Diffusers."
         )
 
@@ -475,7 +475,7 @@ class FlaxFeedForward(nn.Module):
 
     def setup(self):
         logger.warning(
-            "Flax classes are deprecated and will be removed in Diffusers v1.0.0. We "
+            "Flax classes are deprecated and will be removed in Diffusers v0.40.0. We "
             "recommend migrating to PyTorch classes or pinning your version of Diffusers."
         )
 
@@ -510,7 +510,7 @@ class FlaxGEGLU(nn.Module):
 
     def setup(self):
         logger.warning(
-            "Flax classes are deprecated and will be removed in Diffusers v1.0.0. We "
+            "Flax classes are deprecated and will be removed in Diffusers v0.40.0. We "
             "recommend migrating to PyTorch classes or pinning your version of Diffusers."
         )
 
diff --git a/src/diffusers/models/controlnets/controlnet_flax.py b/src/diffusers/models/controlnets/controlnet_flax.py
index 5bc1a446338e..48908695b91e 100644
--- a/src/diffusers/models/controlnets/controlnet_flax.py
+++ b/src/diffusers/models/controlnets/controlnet_flax.py
@@ -52,7 +52,7 @@ class FlaxControlNetConditioningEmbedding(nn.Module):
 
     def setup(self) -> None:
         logger.warning(
-            "Flax classes are deprecated and will be removed in Diffusers v1.0.0. We "
+            "Flax classes are deprecated and will be removed in Diffusers v0.40.0. We "
             "recommend migrating to PyTorch classes or pinning your version of Diffusers."
         )
 
@@ -191,7 +191,7 @@ def init_weights(self, rng: jax.Array) -> FrozenDict:
 
     def setup(self) -> None:
         logger.warning(
-            "Flax classes are deprecated and will be removed in Diffusers v1.0.0. We "
+            "Flax classes are deprecated and will be removed in Diffusers v0.40.0. We "
             "recommend migrating to PyTorch classes or pinning your version of Diffusers."
         )
 
diff --git a/src/diffusers/models/embeddings_flax.py b/src/diffusers/models/embeddings_flax.py
index 3790905e583c..c0e74d5cc8b1 100644
--- a/src/diffusers/models/embeddings_flax.py
+++ b/src/diffusers/models/embeddings_flax.py
@@ -82,7 +82,7 @@ class FlaxTimestepEmbedding(nn.Module):
     """
 
     logger.warning(
-        "Flax classes are deprecated and will be removed in Diffusers v1.0.0. We "
+        "Flax classes are deprecated and will be removed in Diffusers v0.40.0. We "
         "recommend migrating to PyTorch classes or pinning your version of Diffusers."
     )
 
@@ -115,7 +115,7 @@ class FlaxTimesteps(nn.Module):
     freq_shift: float = 1
 
     logger.warning(
-        "Flax classes are deprecated and will be removed in Diffusers v1.0.0. We "
+        "Flax classes are deprecated and will be removed in Diffusers v0.40.0. We "
         "recommend migrating to PyTorch classes or pinning your version of Diffusers."
     )
 
diff --git a/src/diffusers/models/modeling_flax_utils.py b/src/diffusers/models/modeling_flax_utils.py
index 9f62bd7199e0..3bc68172a23b 100644
--- a/src/diffusers/models/modeling_flax_utils.py
+++ b/src/diffusers/models/modeling_flax_utils.py
@@ -285,7 +285,7 @@ def from_pretrained(
         ```
         """
         logger.warning(
-            "Flax classes are deprecated and will be removed in Diffusers v1.0.0. We "
+            "Flax classes are deprecated and will be removed in Diffusers v0.40.0. We "
             "recommend migrating to PyTorch classes or pinning your version of Diffusers."
         )
         config = kwargs.pop("config", None)
diff --git a/src/diffusers/models/resnet_flax.py b/src/diffusers/models/resnet_flax.py
index 9bedaa9a36b6..bd6912bc790a 100644
--- a/src/diffusers/models/resnet_flax.py
+++ b/src/diffusers/models/resnet_flax.py
@@ -27,7 +27,7 @@ class FlaxUpsample2D(nn.Module):
 
     def setup(self):
         logger.warning(
-            "Flax classes are deprecated and will be removed in Diffusers v1.0.0. We "
+            "Flax classes are deprecated and will be removed in Diffusers v0.40.0. We "
             "recommend migrating to PyTorch classes or pinning your version of Diffusers."
         )
 
@@ -56,7 +56,7 @@ class FlaxDownsample2D(nn.Module):
 
     def setup(self):
         logger.warning(
-            "Flax classes are deprecated and will be removed in Diffusers v1.0.0. We "
+            "Flax classes are deprecated and will be removed in Diffusers v0.40.0. We "
             "recommend migrating to PyTorch classes or pinning your version of Diffusers."
         )
 
@@ -84,7 +84,7 @@ class FlaxResnetBlock2D(nn.Module):
 
     def setup(self):
         logger.warning(
-            "Flax classes are deprecated and will be removed in Diffusers v1.0.0. We "
+            "Flax classes are deprecated and will be removed in Diffusers v0.40.0. We "
             "recommend migrating to PyTorch classes or pinning your version of Diffusers."
         )
 
diff --git a/src/diffusers/models/unets/unet_2d_blocks_flax.py b/src/diffusers/models/unets/unet_2d_blocks_flax.py
index 6e6005afdc31..9d0fba45413d 100644
--- a/src/diffusers/models/unets/unet_2d_blocks_flax.py
+++ b/src/diffusers/models/unets/unet_2d_blocks_flax.py
@@ -65,7 +65,7 @@ class FlaxCrossAttnDownBlock2D(nn.Module):
 
     def setup(self):
         logger.warning(
-            "Flax classes are deprecated and will be removed in Diffusers v1.0.0. We "
+            "Flax classes are deprecated and will be removed in Diffusers v0.40.0. We "
             "recommend migrating to PyTorch classes or pinning your version of Diffusers."
         )
 
@@ -145,7 +145,7 @@ class FlaxDownBlock2D(nn.Module):
 
     def setup(self):
         logger.warning(
-            "Flax classes are deprecated and will be removed in Diffusers v1.0.0. We "
+            "Flax classes are deprecated and will be removed in Diffusers v0.40.0. We "
             "recommend migrating to PyTorch classes or pinning your version of Diffusers."
         )
 
@@ -223,7 +223,7 @@ class FlaxCrossAttnUpBlock2D(nn.Module):
 
     def setup(self):
         logger.warning(
-            "Flax classes are deprecated and will be removed in Diffusers v1.0.0. We "
+            "Flax classes are deprecated and will be removed in Diffusers v0.40.0. We "
             "recommend migrating to PyTorch classes or pinning your version of Diffusers."
         )
 
@@ -308,7 +308,7 @@ class FlaxUpBlock2D(nn.Module):
 
     def setup(self):
         logger.warning(
-            "Flax classes are deprecated and will be removed in Diffusers v1.0.0. We "
+            "Flax classes are deprecated and will be removed in Diffusers v0.40.0. We "
             "recommend migrating to PyTorch classes or pinning your version of Diffusers."
         )
 
@@ -381,7 +381,7 @@ class FlaxUNetMidBlock2DCrossAttn(nn.Module):
 
     def setup(self):
         logger.warning(
-            "Flax classes are deprecated and will be removed in Diffusers v1.0.0. We "
+            "Flax classes are deprecated and will be removed in Diffusers v0.40.0. We "
             "recommend migrating to PyTorch classes or pinning your version of Diffusers."
         )
 
diff --git a/src/diffusers/models/unets/unet_2d_condition_flax.py b/src/diffusers/models/unets/unet_2d_condition_flax.py
index a361026fc0ea..e8ba53c83572 100644
--- a/src/diffusers/models/unets/unet_2d_condition_flax.py
+++ b/src/diffusers/models/unets/unet_2d_condition_flax.py
@@ -166,7 +166,7 @@ def init_weights(self, rng: jax.Array) -> FrozenDict:
 
     def setup(self) -> None:
         logger.warning(
-            "Flax classes are deprecated and will be removed in Diffusers v1.0.0. We "
+            "Flax classes are deprecated and will be removed in Diffusers v0.40.0. We "
             "recommend migrating to PyTorch classes or pinning your version of Diffusers."
         )
 
diff --git a/src/diffusers/models/vae_flax.py b/src/diffusers/models/vae_flax.py
index c7042840e4e0..c357f82eadc1 100644
--- a/src/diffusers/models/vae_flax.py
+++ b/src/diffusers/models/vae_flax.py
@@ -76,7 +76,7 @@ class FlaxUpsample2D(nn.Module):
 
     def setup(self):
         logger.warning(
-            "Flax classes are deprecated and will be removed in Diffusers v1.0.0. We "
+            "Flax classes are deprecated and will be removed in Diffusers v0.40.0. We "
             "recommend migrating to PyTorch classes or pinning your version of Diffusers."
         )
         self.conv = nn.Conv(
@@ -114,7 +114,7 @@ class FlaxDownsample2D(nn.Module):
 
     def setup(self):
         logger.warning(
-            "Flax classes are deprecated and will be removed in Diffusers v1.0.0. We "
+            "Flax classes are deprecated and will be removed in Diffusers v0.40.0. We "
             "recommend migrating to PyTorch classes or pinning your version of Diffusers."
         )
 
@@ -161,7 +161,7 @@ class FlaxResnetBlock2D(nn.Module):
 
     def setup(self):
         logger.warning(
-            "Flax classes are deprecated and will be removed in Diffusers v1.0.0. We "
+            "Flax classes are deprecated and will be removed in Diffusers v0.40.0. We "
             "recommend migrating to PyTorch classes or pinning your version of Diffusers."
         )
 
@@ -238,7 +238,7 @@ class FlaxAttentionBlock(nn.Module):
 
     def setup(self):
         logger.warning(
-            "Flax classes are deprecated and will be removed in Diffusers v1.0.0. We "
+            "Flax classes are deprecated and will be removed in Diffusers v0.40.0. We "
             "recommend migrating to PyTorch classes or pinning your version of Diffusers."
         )
 
@@ -324,7 +324,7 @@ class FlaxDownEncoderBlock2D(nn.Module):
 
     def setup(self):
         logger.warning(
-            "Flax classes are deprecated and will be removed in Diffusers v1.0.0. We "
+            "Flax classes are deprecated and will be removed in Diffusers v0.40.0. We "
             "recommend migrating to PyTorch classes or pinning your version of Diffusers."
         )
 
@@ -386,7 +386,7 @@ class FlaxUpDecoderBlock2D(nn.Module):
 
     def setup(self):
         logger.warning(
-            "Flax classes are deprecated and will be removed in Diffusers v1.0.0. We "
+            "Flax classes are deprecated and will be removed in Diffusers v0.40.0. We "
             "recommend migrating to PyTorch classes or pinning your version of Diffusers."
         )
 
@@ -445,7 +445,7 @@ class FlaxUNetMidBlock2D(nn.Module):
 
     def setup(self):
         logger.warning(
-            "Flax classes are deprecated and will be removed in Diffusers v1.0.0. We "
+            "Flax classes are deprecated and will be removed in Diffusers v0.40.0. We "
             "recommend migrating to PyTorch classes or pinning your version of Diffusers."
         )
 
@@ -541,7 +541,7 @@ class FlaxEncoder(nn.Module):
 
     def setup(self):
         logger.warning(
-            "Flax classes are deprecated and will be removed in Diffusers v1.0.0. We "
+            "Flax classes are deprecated and will be removed in Diffusers v0.40.0. We "
             "recommend migrating to PyTorch classes or pinning your version of Diffusers."
         )
 
@@ -658,7 +658,7 @@ class FlaxDecoder(nn.Module):
 
     def setup(self):
         logger.warning(
-            "Flax classes are deprecated and will be removed in Diffusers v1.0.0. We "
+            "Flax classes are deprecated and will be removed in Diffusers v0.40.0. We "
             "recommend migrating to PyTorch classes or pinning your version of Diffusers."
         )
 
@@ -835,7 +835,7 @@ class FlaxAutoencoderKL(nn.Module, FlaxModelMixin, ConfigMixin):
 
     def setup(self):
         logger.warning(
-            "Flax classes are deprecated and will be removed in Diffusers v1.0.0. We "
+            "Flax classes are deprecated and will be removed in Diffusers v0.40.0. We "
             "recommend migrating to PyTorch classes or pinning your version of Diffusers."
         )
 
diff --git a/src/diffusers/pipelines/pipeline_flax_utils.py b/src/diffusers/pipelines/pipeline_flax_utils.py
index 1a87289450eb..8c29db8aa45e 100644
--- a/src/diffusers/pipelines/pipeline_flax_utils.py
+++ b/src/diffusers/pipelines/pipeline_flax_utils.py
@@ -309,7 +309,7 @@ def from_pretrained(cls, pretrained_model_name_or_path: str | os.PathLike | None
         ```
         """
         logger.warning(
-            "Flax classes are deprecated and will be removed in Diffusers v1.0.0. We "
+            "Flax classes are deprecated and will be removed in Diffusers v0.40.0. We "
             "recommend migrating to PyTorch classes or pinning your version of Diffusers."
         )
 
diff --git a/src/diffusers/schedulers/scheduling_ddim_flax.py b/src/diffusers/schedulers/scheduling_ddim_flax.py
index 3bb44caef2bf..45c173ffbff5 100644
--- a/src/diffusers/schedulers/scheduling_ddim_flax.py
+++ b/src/diffusers/schedulers/scheduling_ddim_flax.py
@@ -129,7 +129,7 @@ def __init__(
         dtype: jnp.dtype = jnp.float32,
     ):
         logger.warning(
-            "Flax classes are deprecated and will be removed in Diffusers v1.0.0. We "
+            "Flax classes are deprecated and will be removed in Diffusers v0.40.0. We "
             "recommend migrating to PyTorch classes or pinning your version of Diffusers."
         )
         self.dtype = dtype
diff --git a/src/diffusers/schedulers/scheduling_ddpm_flax.py b/src/diffusers/schedulers/scheduling_ddpm_flax.py
index 7840ff729488..b286999bcf3c 100644
--- a/src/diffusers/schedulers/scheduling_ddpm_flax.py
+++ b/src/diffusers/schedulers/scheduling_ddpm_flax.py
@@ -114,7 +114,7 @@ def __init__(
         dtype: jnp.dtype = jnp.float32,
     ):
         logger.warning(
-            "Flax classes are deprecated and will be removed in Diffusers v1.0.0. We "
+            "Flax classes are deprecated and will be removed in Diffusers v0.40.0. We "
             "recommend migrating to PyTorch classes or pinning your version of Diffusers."
         )
         self.dtype = dtype
diff --git a/src/diffusers/schedulers/scheduling_dpmsolver_multistep_flax.py b/src/diffusers/schedulers/scheduling_dpmsolver_multistep_flax.py
index dd579e84d609..e8b5f6673037 100644
--- a/src/diffusers/schedulers/scheduling_dpmsolver_multistep_flax.py
+++ b/src/diffusers/schedulers/scheduling_dpmsolver_multistep_flax.py
@@ -175,7 +175,7 @@ def __init__(
         dtype: jnp.dtype = jnp.float32,
     ):
         logger.warning(
-            "Flax classes are deprecated and will be removed in Diffusers v1.0.0. We "
+            "Flax classes are deprecated and will be removed in Diffusers v0.40.0. We "
             "recommend migrating to PyTorch classes or pinning your version of Diffusers."
         )
         self.dtype = dtype
diff --git a/src/diffusers/schedulers/scheduling_euler_discrete_flax.py b/src/diffusers/schedulers/scheduling_euler_discrete_flax.py
index fc555dd21395..91150c5cddd8 100644
--- a/src/diffusers/schedulers/scheduling_euler_discrete_flax.py
+++ b/src/diffusers/schedulers/scheduling_euler_discrete_flax.py
@@ -112,7 +112,7 @@ def __init__(
         dtype: jnp.dtype = jnp.float32,
     ):
         logger.warning(
-            "Flax classes are deprecated and will be removed in Diffusers v1.0.0. We "
+            "Flax classes are deprecated and will be removed in Diffusers v0.40.0. We "
             "recommend migrating to PyTorch classes or pinning your version of Diffusers."
         )
         self.dtype = dtype
diff --git a/src/diffusers/schedulers/scheduling_karras_ve_flax.py b/src/diffusers/schedulers/scheduling_karras_ve_flax.py
index 04c46220fcca..af7c17f17cf0 100644
--- a/src/diffusers/schedulers/scheduling_karras_ve_flax.py
+++ b/src/diffusers/schedulers/scheduling_karras_ve_flax.py
@@ -105,7 +105,7 @@ def __init__(
         s_max: float = 50,
     ):
         logger.warning(
-            "Flax classes are deprecated and will be removed in Diffusers v1.0.0. We "
+            "Flax classes are deprecated and will be removed in Diffusers v0.40.0. We "
             "recommend migrating to PyTorch classes or pinning your version of Diffusers."
         )
 
diff --git a/src/diffusers/schedulers/scheduling_lms_discrete_flax.py b/src/diffusers/schedulers/scheduling_lms_discrete_flax.py
index 65902678e1d9..c37d8752f7fb 100644
--- a/src/diffusers/schedulers/scheduling_lms_discrete_flax.py
+++ b/src/diffusers/schedulers/scheduling_lms_discrete_flax.py
@@ -114,7 +114,7 @@ def __init__(
         dtype: jnp.dtype = jnp.float32,
     ):
         logger.warning(
-            "Flax classes are deprecated and will be removed in Diffusers v1.0.0. We "
+            "Flax classes are deprecated and will be removed in Diffusers v0.40.0. We "
             "recommend migrating to PyTorch classes or pinning your version of Diffusers."
         )
         self.dtype = dtype
diff --git a/src/diffusers/schedulers/scheduling_pndm_flax.py b/src/diffusers/schedulers/scheduling_pndm_flax.py
index 377e2d35ba0a..e18e484c8c4c 100644
--- a/src/diffusers/schedulers/scheduling_pndm_flax.py
+++ b/src/diffusers/schedulers/scheduling_pndm_flax.py
@@ -135,7 +135,7 @@ def __init__(
         dtype: jnp.dtype = jnp.float32,
     ):
         logger.warning(
-            "Flax classes are deprecated and will be removed in Diffusers v1.0.0. We "
+            "Flax classes are deprecated and will be removed in Diffusers v0.40.0. We "
             "recommend migrating to PyTorch classes or pinning your version of Diffusers."
         )
         self.dtype = dtype
diff --git a/src/diffusers/schedulers/scheduling_sde_ve_flax.py b/src/diffusers/schedulers/scheduling_sde_ve_flax.py
index 62a54c7dc948..59c3a1cbaeeb 100644
--- a/src/diffusers/schedulers/scheduling_sde_ve_flax.py
+++ b/src/diffusers/schedulers/scheduling_sde_ve_flax.py
@@ -103,7 +103,7 @@ def __init__(
         correct_steps: int = 1,
     ):
         logger.warning(
-            "Flax classes are deprecated and will be removed in Diffusers v1.0.0. We "
+            "Flax classes are deprecated and will be removed in Diffusers v0.40.0. We "
             "recommend migrating to PyTorch classes or pinning your version of Diffusers."
         )
 
diff --git a/src/diffusers/schedulers/scheduling_utils_flax.py b/src/diffusers/schedulers/scheduling_utils_flax.py
index 44de56a3980d..f05cea3227df 100644
--- a/src/diffusers/schedulers/scheduling_utils_flax.py
+++ b/src/diffusers/schedulers/scheduling_utils_flax.py
@@ -128,7 +128,7 @@ def from_pretrained(
 
         """
         logger.warning(
-            "Flax classes are deprecated and will be removed in Diffusers v1.0.0. We "
+            "Flax classes are deprecated and will be removed in Diffusers v0.40.0. We "
             "recommend migrating to PyTorch classes or pinning your version of Diffusers."
         )
         config, kwargs = cls.load_config(