premerge 1

This commit is contained in:
Seunghoon Lee
2024-02-01 01:20:16 +09:00
parent df6d6e17f5
commit c26c53a9ad
4 changed files with 31 additions and 38 deletions
+2 -1
View File
@@ -3,6 +3,7 @@ import time
import math
import inspect
import typing
import numpy as np
import torch
import torchvision.transforms.functional as TF
import diffusers
@@ -475,7 +476,7 @@ def process_diffusers(p: processing.StableDiffusionProcessing):
return max(1, int(steps))
shared.sd_model = update_pipeline(shared.sd_model, p)
onnx_preprocess_pipeline(p, is_refiner_enabled())
preprocess_onnx_pipeline(p, is_refiner_enabled())
base_args = set_pipeline_args(
model=shared.sd_model,
prompts=p.prompts,