|
37 | 37 | from invokeai.backend.model_manager.config import (
|
38 | 38 | AnyModelConfig,
|
39 | 39 | CheckpointConfigBase,
|
40 |
| - CLIPEmbedCheckpointConfig, |
41 | 40 | CLIPEmbedDiffusersConfig,
|
42 | 41 | ControlNetCheckpointConfig,
|
43 | 42 | ControlNetDiffusersConfig,
|
@@ -131,27 +130,6 @@ def _load_model(
|
131 | 130 | )
|
132 | 131 |
|
133 | 132 |
|
134 |
| -@ModelLoaderRegistry.register(base=BaseModelType.Any, type=ModelType.CLIPEmbed, format=ModelFormat.Checkpoint) |
135 |
| -class CLIPCheckpointLoader(ModelLoader): |
136 |
| - """Class to load main models.""" |
137 |
| - |
138 |
| - def _load_model( |
139 |
| - self, |
140 |
| - config: AnyModelConfig, |
141 |
| - submodel_type: Optional[SubModelType] = None, |
142 |
| - ) -> AnyModel: |
143 |
| - if not isinstance(config, CLIPEmbedCheckpointConfig): |
144 |
| - raise ValueError("Only CLIPEmbedCheckpointConfig models are currently supported here.") |
145 |
| - |
146 |
| - match submodel_type: |
147 |
| - case SubModelType.TextEncoder: |
148 |
| - return CLIPTextModel.from_pretrained(Path(config.path), use_safetensors=True) |
149 |
| - case _: |
150 |
| - raise ValueError( |
151 |
| - f"Only TextEncoder submodels are currently supported. Received: {submodel_type.value if submodel_type else 'None'}" |
152 |
| - ) |
153 |
| - |
154 |
| - |
155 | 133 | @ModelLoaderRegistry.register(base=BaseModelType.Any, type=ModelType.T5Encoder, format=ModelFormat.BnbQuantizedLlmInt8b)
|
156 | 134 | class BnbQuantizedLlmInt8bCheckpointModel(ModelLoader):
|
157 | 135 | """Class to load main models."""
|
|
0 commit comments