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)
|