From 446b0b69893bb12f304588023e8c29eeeea4774d Mon Sep 17 00:00:00 2001 From: Jaret Burkett Date: Thu, 26 Jun 2025 16:46:58 -0600 Subject: [PATCH] Remove revision for kontext --- .../diffusion_models/flux_kontext/flux_kontext.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) 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)