From 2c1f49e3992eaa00c766ce43e13bf945b058568b Mon Sep 17 00:00:00 2001 From: David Burnett Date: Thu, 17 Oct 2024 13:58:31 +0100 Subject: [PATCH] ruff fix --- invokeai/backend/model_manager/load/model_loaders/flux.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/invokeai/backend/model_manager/load/model_loaders/flux.py b/invokeai/backend/model_manager/load/model_loaders/flux.py index fe9f17eeedb..76eab51ea88 100644 --- a/invokeai/backend/model_manager/load/model_loaders/flux.py +++ b/invokeai/backend/model_manager/load/model_loaders/flux.py @@ -175,7 +175,7 @@ def _load_model( case SubModelType.Tokenizer2: return T5Tokenizer.from_pretrained(Path(config.path) / "tokenizer_2", max_length=512) case SubModelType.TextEncoder2: - return T5EncoderModel.from_pretrained(Path(config.path) / "text_encoder_2", torch_dtype='auto') + return T5EncoderModel.from_pretrained(Path(config.path) / "text_encoder_2", torch_dtype="auto") raise ValueError( f"Only Tokenizer and TextEncoder submodels are currently supported. Received: {submodel_type.value if submodel_type else 'None'}"