mirror of
https://github.com/vladmandic/automatic
synced 2026-09-19 01:04:32 +02:00
check for sfast with sag
This commit is contained in:
@@ -344,7 +344,9 @@ def process_diffusers(p: processing.StableDiffusionProcessing):
|
||||
if p.sag_scale > 0 and is_txt2img():
|
||||
update_sampler(shared.sd_model)
|
||||
supported = ['DDIMScheduler', 'PNDMScheduler', 'DDPMScheduler', 'DEISMultistepScheduler', 'UniPCMultistepScheduler', 'DPMSolverMultistepScheduler', 'DPMSolverSinlgestepScheduler']
|
||||
if sd_model.scheduler.__class__.__name__ in supported:
|
||||
if hasattr(sd_model, 'sfast'):
|
||||
shared.log.warning(f'SAG incompatible compile mode: backend={shared.opts.cuda_compile_backend}')
|
||||
elif sd_model.scheduler.__class__.__name__ in supported:
|
||||
sd_model = sd_models.switch_pipe(diffusers.StableDiffusionSAGPipeline, sd_model)
|
||||
p.extra_generation_params["SAG scale"] = p.sag_scale
|
||||
p.task_args['sag_scale'] = p.sag_scale
|
||||
|
||||
@@ -164,6 +164,7 @@ def compile_stablefast(sd_model):
|
||||
try:
|
||||
t0 = time.time()
|
||||
sd_model = sf.compile(sd_model, config)
|
||||
sd_model.sfast = True
|
||||
setup_logging() # compile messes with logging so reset is needed
|
||||
if shared.opts.cuda_compile_precompile:
|
||||
sd_model("dummy prompt")
|
||||
|
||||
Reference in New Issue
Block a user