diff --git a/extensions_built_in/diffusion_models/flux_kontext/flux_kontext.py b/extensions_built_in/diffusion_models/flux_kontext/flux_kontext.py
index 84806017..07212151 100644
--- a/extensions_built_in/diffusion_models/flux_kontext/flux_kontext.py
+++ b/extensions_built_in/diffusion_models/flux_kontext/flux_kontext.py
@@ -95,8 +95,7 @@ class FluxKontextModel(BaseModel):
         transformer = FluxTransformer2DModel.from_pretrained(
             transformer_path,
             subfolder=transformer_subfolder,
-            torch_dtype=dtype,
-            revision="7610c9af"
+            torch_dtype=dtype
         )
         transformer.to(self.quantize_device, dtype=dtype)