mirror of
https://github.com/vladmandic/automatic
synced 2026-09-20 01:31:13 +02:00
@@ -75,7 +75,7 @@ class ConsiStoryScript(scripts_manager.Script):
|
||||
shared.sd_model = sd_models.switch_pipe(cs.ConsistoryExtendAttnSDXLPipeline, shared.sd_model)
|
||||
shared.sd_model.unet = cs.ConsistorySDXLUNet2DConditionModel.from_config(shared.sd_model.unet.config)
|
||||
shared.sd_model.unet.load_state_dict(state_dict) # now load it into new class
|
||||
shared.sd_model.unet.to(dtype=devices.dtype) # ty: ignore
|
||||
shared.sd_model.unet.to(dtype=devices.dtype)
|
||||
state_dict = None
|
||||
# sd_models.set_diffuser_options(shared.sd_model)
|
||||
sd_models.move_model(shared.sd_model, devices.device)
|
||||
|
||||
@@ -90,7 +90,7 @@ def set_template(
|
||||
pass
|
||||
elif options.processor is None:
|
||||
log.error('Prompt enhance: image not supported by model')
|
||||
return prompt # Return original text part if image cannot be processed
|
||||
return prompt if prompt is not None else '' # Return original text part if image cannot be processed
|
||||
|
||||
if has_image:
|
||||
chat_template = get_image_template(system, prompt, options, nsfw, has_system, has_prompt, has_processor, is_video, image)
|
||||
|
||||
Reference in New Issue
Block a user