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

2
invokeai/app/invocations/onnx.py
View File

@ -212,7 +212,7 @@ class ONNXTextToLatentsInvocation(BaseInvocation):
context=context, context=context,
scheduler_info=self.unet.scheduler, scheduler_info=self.unet.scheduler,
scheduler_name=self.scheduler, scheduler_name=self.scheduler,
seed=0, # TODO: refactor this node seed=0, # TODO: refactor this node
) )
def torch2numpy(latent: torch.Tensor): def torch2numpy(latent: torch.Tensor):

36
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(
step=-1, PipelineIntermediateState(
order=self.scheduler.order, step=-1,
total_steps=len(timesteps), order=self.scheduler.order,
timestep=self.scheduler.config.num_train_timesteps, total_steps=len(timesteps),
latents=latents, timestep=self.scheduler.config.num_train_timesteps,
)) 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,15 +471,17 @@ 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(
step=i, PipelineIntermediateState(
order=self.scheduler.order, step=i,
total_steps=len(timesteps), order=self.scheduler.order,
timestep=int(t), total_steps=len(timesteps),
latents=latents, timestep=int(t),
predicted_original=predicted_original, latents=latents,
attention_map_saver=attention_map_saver, predicted_original=predicted_original,
)) 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,
)