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

Fix typing of timesteps and init_timestep.

Browse Source
This commit is contained in:
Ryan Dick 2024-06-12 15:11:24 -04:00
parent c2cf1137e9
commit 3aef717ef4
2 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -601,7 +601,7 @@ class DenoiseLatentsInvocation(BaseInvocation):
denoising_start: float, denoising_start: float,
denoising_end: float, denoising_end: float,
seed: int, seed: int,
) -> Tuple[List[int], int, Dict[str, Any]]: ) -> Tuple[torch.Tensor, torch.Tensor, Dict[str, Any]]:
assert isinstance(scheduler, ConfigMixin) assert isinstance(scheduler, ConfigMixin)
if scheduler.config.get("cpu_only", False): if scheduler.config.get("cpu_only", False):
scheduler.set_timesteps(steps, device="cpu") scheduler.set_timesteps(steps, device="cpu")

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

@ -360,7 +360,7 @@ class StableDiffusionGeneratorPipeline(StableDiffusionPipeline):
def generate_latents_from_embeddings( def generate_latents_from_embeddings(
self, self,
latents: torch.Tensor, latents: torch.Tensor,
timesteps, timesteps: torch.Tensor,
conditioning_data: TextConditioningData, conditioning_data: TextConditioningData,
scheduler_step_kwargs: dict[str, Any], scheduler_step_kwargs: dict[str, Any],
callback: Callable[[PipelineIntermediateState], None], callback: Callable[[PipelineIntermediateState], None],