From bbf934d9805501d246aa7434c985c7c525999ec5 Mon Sep 17 00:00:00 2001 From: Ryan Dick Date: Mon, 26 Aug 2024 15:23:35 +0000 Subject: [PATCH] Remove outdated TODO. --- invokeai/backend/model_manager/load/model_loaders/flux.py | 4 +--- 1 file changed, 1 insertion(+), 3 deletions(-) diff --git a/invokeai/backend/model_manager/load/model_loaders/flux.py b/invokeai/backend/model_manager/load/model_loaders/flux.py index 33090ccc10..53119f6df0 100644 --- a/invokeai/backend/model_manager/load/model_loaders/flux.py +++ b/invokeai/backend/model_manager/load/model_loaders/flux.py @@ -153,9 +153,7 @@ class T5EncoderCheckpointModel(ModelLoader): 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" - ) # TODO: Fix hf subfolder install + return T5EncoderModel.from_pretrained(Path(config.path) / "text_encoder_2") raise ValueError( f"Only Tokenizer and TextEncoder submodels are currently supported. Received: {submodel_type.value if submodel_type else 'None'}"