mirror of
https://github.com/invoke-ai/InvokeAI
synced 2024-08-30 20:32:17 +00:00
remove factory pattern
Factory pattern is now removed. Typical usage of the InvokeAIGenerator is now: ``` from invokeai.backend.generator import ( InvokeAIGeneratorBasicParams, Txt2Img, Img2Img, Inpaint, ) params = InvokeAIGeneratorBasicParams( model_name = 'stable-diffusion-1.5', steps = 30, scheduler = 'k_lms', cfg_scale = 8.0, height = 640, width = 640 ) print ('=== TXT2IMG TEST ===') txt2img = Txt2Img(manager, params) outputs = txt2img.generate(prompt='banana sushi', iterations=2) for i in outputs: print(f'image={output.image}, seed={output.seed}, model={output.params.model_name}, hash={output.model_hash}, steps={output.params.steps}') ``` The `params` argument is optional, so if you wish to accept default parameters and selectively override them, just do this: ``` outputs = Txt2Img(manager).generate(prompt='banana sushi', steps=50, scheduler='k_heun', model_name='stable-diffusion-2.1' ) ```
This commit is contained in:
@ -34,7 +34,7 @@ from picklescan.scanner import scan_file_path
|
||||
from invokeai.backend.globals import Globals, global_cache_dir
|
||||
|
||||
from ..stable_diffusion import StableDiffusionGeneratorPipeline
|
||||
from ..util import CUDA_DEVICE, ask_user, download_with_resume
|
||||
from ..util import CUDA_DEVICE, CPU_DEVICE, ask_user, download_with_resume
|
||||
|
||||
class SDLegacyType(Enum):
|
||||
V1 = 1
|
||||
|
Reference in New Issue
Block a user