Mirrored_Repos/InvokeAI
1
0
Fork 0
mirror of https://github.com/invoke-ai/InvokeAI synced 2024-08-30 20:32:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

chore: black

Browse Source
This commit is contained in:
psychedelicious 2023-08-14 13:02:33 +10:00
parent 46a8eed33e
commit 9d3cd85bdd
3 changed files with 27 additions and 18 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
invokeai/backend/stable_diffusion/diffusers_pipeline.py
View File

@ -429,13 +429,15 @@ class StableDiffusionGeneratorPipeline(StableDiffusionPipeline):
step_count=len(self.scheduler.timesteps), step_count=len(self.scheduler.timesteps),
): ):
if callback is not None: if callback is not None:
callback(PipelineIntermediateState( callback(
PipelineIntermediateState(
step=-1, step=-1,
order=self.scheduler.order, order=self.scheduler.order,
total_steps=len(timesteps), total_steps=len(timesteps),
timestep=self.scheduler.config.num_train_timesteps, timestep=self.scheduler.config.num_train_timesteps,
latents=latents, latents=latents,
)) )
)
# print("timesteps:", timesteps) # print("timesteps:", timesteps)
for i, t in enumerate(self.progress_bar(timesteps)): for i, t in enumerate(self.progress_bar(timesteps)):
@ -469,7 +471,8 @@ class StableDiffusionGeneratorPipeline(StableDiffusionPipeline):
# self.invokeai_diffuser.setup_attention_map_saving(attention_map_saver) # self.invokeai_diffuser.setup_attention_map_saving(attention_map_saver)
if callback is not None: if callback is not None:
callback(PipelineIntermediateState( callback(
PipelineIntermediateState(
step=i, step=i,
order=self.scheduler.order, order=self.scheduler.order,
total_steps=len(timesteps), total_steps=len(timesteps),
@ -477,7 +480,8 @@ class StableDiffusionGeneratorPipeline(StableDiffusionPipeline):
latents=latents, latents=latents,
predicted_original=predicted_original, predicted_original=predicted_original,
attention_map_saver=attention_map_saver, attention_map_saver=attention_map_saver,
)) )
)
return latents, attention_map_saver return latents, attention_map_saver

7
invokeai/backend/stable_diffusion/diffusion/__init__.py
View File

@ -3,4 +3,9 @@ Initialization file for invokeai.models.diffusion
""" """
from .cross_attention_control import InvokeAICrossAttentionMixin from .cross_attention_control import InvokeAICrossAttentionMixin
from .cross_attention_map_saving import AttentionMapSaver from .cross_attention_map_saving import AttentionMapSaver
from .shared_invokeai_diffusion import InvokeAIDiffuserComponent, PostprocessingSettings, BasicConditioningInfo, SDXLConditioningInfo from .shared_invokeai_diffusion import (
InvokeAIDiffuserComponent,
PostprocessingSettings,
BasicConditioningInfo,
SDXLConditioningInfo,
)