Public/automatic
1
0
Fork 0
mirror of https://github.com/vladmandic/automatic synced 2026-08-26 15:16:01 +02:00
Code Issues Packages Projects Releases Wiki Activity

refactor onnx and olive

Browse Source
This commit is contained in:
Seunghoon Lee
2023-11-04 23:46:20 +09:00
parent b70258c926
commit 6507491d8f
27 changed files with 1185 additions and 342 deletions
Download Patch File Download Diff File Expand all files Collapse all files
modules/processing_diffusers.py
+2 -2
View File
@@ -21,8 +21,8 @@ def process_diffusers(p: processing.StableDiffusionProcessing):
orig_pipeline = shared.sd_model
results = []
if isinstance(shared.sd_model, OlivePipeline):
shared.sd_model = shared.sd_model.optimize(p.width, p.height)
if hasattr(shared.sd_model, 'preprocess'):
shared.sd_model = shared.sd_model.preprocess(p.width, p.height, p.batch_size)
def is_txt2img():
return sd_models.get_diffusers_task(shared.sd_model) == sd_models.DiffusersTaskType.TEXT_2_IMAGE