omer11a commited on
Commit
fa4d90e
·
1 Parent(s): 33651e9

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)