Public/automatic
1
0
Fork 0
mirror of https://github.com/vladmandic/automatic synced 2026-09-19 01:04:32 +02:00
Code Issues Packages Projects Releases Wiki Activity

add sdxl-turbo

Browse Source
This commit is contained in:
Vladimir Mandic
2023-11-29 16:32:12 -05:00
parent d8d8f58072
commit f08b4e5c23
10 changed files with 39 additions and 31 deletions
Download Patch File Download Diff File Expand all files Collapse all files
scripts/animatediff.py
+6 -8
View File
@@ -52,9 +52,6 @@ def set_adapter(name: str = None):
if shared.backend != shared.Backend.DIFFUSERS:
shared.log.warning('AnimateDiff: not in diffusers mode')
return
if shared.sd_model_type != 'sd':
shared.log.warning(f'AnimateDiff: unsupported model type: {shared.sd_model.__class__.__name__}')
return
global motion_adapter, loaded_adapter, orig_pipe # pylint: disable=global-statement
adapter_name = name if name is not None and isinstance(name, str) else loaded_adapter
if adapter_name is None or adapter_name == 'None' or shared.sd_model is None:
@@ -65,6 +62,9 @@ def set_adapter(name: str = None):
shared.sd_model = orig_pipe
orig_pipe = None
return
if shared.sd_model_type != 'sd':
shared.log.warning(f'AnimateDiff: unsupported model type: {shared.sd_model.__class__.__name__}')
return
if motion_adapter is not None and loaded_adapter == adapter_name:
shared.log.info(f'AnimateDiff cache: adapter="{adapter_name}"')
return
@@ -139,11 +139,9 @@ class Script(scripts.Script):
shared.sd_model.set_adapters([lora], adapter_weights=[strength])
p.extra_generation_params['AnimateDiff Lora'] = f'{lora}:{strength}'
p.do_not_save_grid = True
p.task_args = {
'num_frames': frames,
'output_type': 'np', # TODO: AnimateDiff use latents and update vae_decode
'num_inference_steps': p.steps,
}
p.task_args['num_frames'] = frames
p.task_args['output_type'] = 'np' # TODO: AnimateDiff use latents and update vae_decode
p.task_args['num_inference_steps'] = p.steps
def postprocess(self, p: processing.StableDiffusionProcessing, processed: processing.Processed, adapter_index, frames, lora_index, strength, override, create_gif, duration, loop): # pylint: disable=arguments-differ, unused-argument
if not create_gif or len(processed.images) < 2:
scripts/ipadapter.py
+1 -1
View File
@@ -106,5 +106,5 @@ class Script(scripts.Script):
else:
shared.log.debug(f'IP adapter cache: adapter="{model}" scale={scale} image={image}')
shared.sd_model.set_ip_adapter_scale(scale)
p.task_args = { 'ip_adapter_image': p.batch_size * [image] }
p.task_args['ip_adapter_image'] = p.batch_size * [image]
p.extra_generation_params["IP Adapter"] = f'{adapter}:{scale}'