Fix typing of timesteps and init_timestep.

invokeai/app/invocations/denoise_latents.py

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

invokeai/backend/stable_diffusion/diffusers_pipeline.py

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