mirror of
https://github.com/invoke-ai/InvokeAI
synced 2024-08-30 20:32:17 +00:00
d44151d6ff
- New method is ModelManager.get_sub_model(model_name:str,model_part:SDModelComponent)
To use:
```
from invokeai.backend import ModelManager, SDModelComponent as sdmc
manager = ModelManager('/path/to/models.yaml')
vae = manager.get_sub_model('stable-diffusion-1.5', sdmc.vae)
```
11 lines
254 B
Python
11 lines
254 B
Python
"""
|
|
Initialization file for invokeai.backend.model_management
|
|
"""
|
|
from .convert_ckpt_to_diffusers import (
|
|
convert_ckpt_to_diffusers,
|
|
load_pipeline_from_original_stable_diffusion_ckpt,
|
|
)
|
|
from .model_manager import ModelManager,SDModelComponent
|
|
|
|
|