Compare commits

...

1 Commits

Author SHA1 Message Date
Patrick von Platen
dc8856e2e1 [WIP] Don't merge / CPU offload example 2023-01-10 16:58:22 +00:00

View File

@@ -500,6 +500,7 @@ class StableDiffusionPipeline(DiffusionPipeline):
text_embeddings = self._encode_prompt(
prompt, device, num_images_per_prompt, do_classifier_free_guidance, negative_prompt
)
torch.cuda.empty_cache()
# 4. Prepare timesteps
self.scheduler.set_timesteps(num_inference_steps, device=device)