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'}"