fix python 3.9 compatibility

This commit is contained in:
Vladimir Mandic
2023-08-23 08:18:40 +00:00
parent 3d1bcf88a2
commit 84f343687d
2 changed files with 6 additions and 7 deletions
+2 -2
View File
@@ -34,7 +34,7 @@ CLIP_SKIP_MAPPING = {
def compel_encode_prompts(
pipeline: diffusers.StableDiffusionXLPipeline | diffusers.StableDiffusionPipeline,
pipeline,
prompts: list,
negative_prompts: list,
prompts_2: typing.Optional[list] = None,
@@ -67,7 +67,7 @@ def compel_encode_prompts(
def compel_encode_prompt(
pipeline: diffusers.StableDiffusionXLPipeline | diffusers.StableDiffusionPipeline,
pipeline,
prompt: str,
negative_prompt: str,
prompt_2: typing.Optional[str] = None,
+4 -5
View File
@@ -864,10 +864,10 @@ class DiffusersTaskType(Enum):
INPAINTING = 3
def set_diffuser_pipe(pipe, new_pipe_type):
sd_checkpoint_info = pipe.sd_checkpoint_info
sd_model_checkpoint = pipe.sd_model_checkpoint
sd_model_hash = pipe.sd_model_hash
has_accelerate = pipe.has_accelerate
sd_checkpoint_info = pipe.sd_checkpoint_info if hasattr(pipe, "sd_checkpoint_info") else None
sd_model_checkpoint = pipe.sd_model_checkpoint if hasattr(pipe, "sd_model_checkpoint") else None
sd_model_hash = pipe.sd_model_hash if hasattr(pipe, "sd_model_hash") else None
has_accelerate = pipe.has_accelerate if hasattr(pipe, "has_accelerate") else None
if new_pipe_type == DiffusersTaskType.TEXT_2_IMAGE:
new_pipe = diffusers.AutoPipelineForText2Image.from_pipe(pipe)
@@ -882,7 +882,6 @@ def set_diffuser_pipe(pipe, new_pipe_type):
new_pipe.sd_model_checkpoint = sd_model_checkpoint
new_pipe.sd_model_hash = sd_model_hash
new_pipe.has_accelerate = has_accelerate
model_data.sd_model = new_pipe
shared.log.info(f"Pipeline class changed from {pipe.__class__.__name__} to {new_pipe.__class__.__name__}")