[Bugfix] prevent cli crash (#3132)

Prevent legacy CLI crash caused by removal of convert option
    
- Compensatory change to the CLI that prevents it from crashing when it
tries to import a model.
- Bug introduced when the "convert" option removed from the model
manager.
This commit is contained in:
Lincoln Stein 2023-04-25 03:55:33 +01:00 committed by GitHub
commit c2a0e8afc3
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 3 additions and 2 deletions

View File

@ -10,7 +10,7 @@ from .generator import (
Img2Img, Img2Img,
Inpaint Inpaint
) )
from .model_management import ModelManager from .model_management import ModelManager, SDModelComponent
from .safety_checker import SafetyChecker from .safety_checker import SafetyChecker
from .args import Args from .args import Args
from .globals import Globals from .globals import Globals

View File

@ -5,6 +5,7 @@ from .convert_ckpt_to_diffusers import (
convert_ckpt_to_diffusers, convert_ckpt_to_diffusers,
load_pipeline_from_original_stable_diffusion_ckpt, load_pipeline_from_original_stable_diffusion_ckpt,
) )
from .model_manager import ModelManager from .model_manager import ModelManager,SDModelComponent