try without setting every time

This commit is contained in:
damian 2023-01-26 17:46:51 +01:00
parent c381788ab9
commit e090c0dc10

View File

@ -366,7 +366,6 @@ class StableDiffusionGeneratorPipeline(StableDiffusionPipeline):
self.scheduler.set_timesteps(num_inference_steps, device=self.unet.device)
timesteps = self.scheduler.timesteps
infer_latents_from_embeddings = GeneratorToCallbackinator(self.generate_latents_from_embeddings, PipelineIntermediateState)
self._enable_memory_efficient_attention()
result: PipelineIntermediateState = infer_latents_from_embeddings(
latents, timesteps, conditioning_data,
noise=noise,