mirror of
https://github.com/invoke-ai/InvokeAI
synced 2024-08-30 20:32:17 +00:00
add a new method to model_manager that retrieves individual pipeline parts
- 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) ```
This commit is contained in:
@ -10,7 +10,7 @@ from .generator import (
|
||||
Img2Img,
|
||||
Inpaint
|
||||
)
|
||||
from .model_management import ModelManager
|
||||
from .model_management import ModelManager, SDModelComponent
|
||||
from .safety_checker import SafetyChecker
|
||||
from .args import Args
|
||||
from .globals import Globals
|
||||
|
Reference in New Issue
Block a user