strict typechecking

Signed-off-by: Vladimir Mandic <mandic00@live.com>
This commit is contained in:
Vladimir Mandic
2026-08-16 11:32:47 +02:00
parent f30e6baa67
commit 76941efc7e
44 changed files with 165 additions and 155 deletions
+1 -1
View File
@@ -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)
+1 -1
View File
@@ -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)