Spaces:
Runtime error
Runtime error
Use full precision
Browse files
pipeline_stable_diffusion_xl_opt.py
CHANGED
@@ -831,8 +831,6 @@ class StableDiffusionXLPipeline(DiffusionPipeline, FromSingleFileMixin, LoraLoad
|
|
831 |
num_inference_steps = len(list(filter(lambda ts: ts >= discrete_timestep_cutoff, timesteps)))
|
832 |
timesteps = timesteps[:num_inference_steps]
|
833 |
|
834 |
-
latents = latents.half()
|
835 |
-
prompt_embeds = prompt_embeds.half()
|
836 |
with self.progress_bar(total=num_inference_steps) as progress_bar:
|
837 |
for i, t in enumerate(timesteps):
|
838 |
latents = self.update_loss(latents, i, t, prompt_embeds, cross_attention_kwargs, add_text_embeds, add_time_ids)
|
|
|
831 |
num_inference_steps = len(list(filter(lambda ts: ts >= discrete_timestep_cutoff, timesteps)))
|
832 |
timesteps = timesteps[:num_inference_steps]
|
833 |
|
|
|
|
|
834 |
with self.progress_bar(total=num_inference_steps) as progress_bar:
|
835 |
for i, t in enumerate(timesteps):
|
836 |
latents = self.update_loss(latents, i, t, prompt_embeds, cross_attention_kwargs, add_text_embeds, add_time_ids)
|