Remove revision for kontext

2025-06-26 16:46:58 -06:00 · 2025-06-26 16:46:58 -06:00 · 446b0b6989
parent 60ef2f1df7
commit 446b0b6989
1 changed files with 1 additions and 2 deletions
--- 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)