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)
```
17 lines
386 B
Python
17 lines
386 B
Python
"""
|
|
Initialization file for invokeai.backend
|
|
"""
|
|
from .generate import Generate
|
|
from .generator import (
|
|
InvokeAIGeneratorBasicParams,
|
|
InvokeAIGenerator,
|
|
InvokeAIGeneratorOutput,
|
|
Txt2Img,
|
|
Img2Img,
|
|
Inpaint
|
|
)
|
|
from .model_management import ModelManager, SDModelComponent
|
|
from .safety_checker import SafetyChecker
|
|
from .args import Args
|
|
from .globals import Globals
|