Disty0
/

zero123plus-pipeline

Model card Files Files and versions

Disty0 commited on Oct 25, 2023

Commit

d639c9b

·

1 Parent(s): 1ad6f7a

device fix

Files changed (1) hide show

pipeline.py +1 -1

pipeline.py CHANGED Viewed

@@ -374,7 +374,7 @@ class Zero123PlusPipeline(diffusers.StableDiffusionPipeline):
                 False,
             )
         ramp = global_embeds.new_tensor(self.config.ramping_coefficients).unsqueeze(-1)
-        encoder_hidden_states = torch.cat([prompt_embeds, torch.zeros(prompt_embeds.shape)]) + global_embeds * ramp
         cak = dict(cond_lat=cond_lat)
         if hasattr(self.unet, "controlnet"):
             cak['control_depth'] = depth_image

                 False,
             )
         ramp = global_embeds.new_tensor(self.config.ramping_coefficients).unsqueeze(-1)
+        encoder_hidden_states = torch.cat([prompt_embeds, torch.zeros(prompt_embeds.shape, device=prompt_embeds.device)]) + global_embeds * ramp
         cak = dict(cond_lat=cond_lat)
         if hasattr(self.unet, "controlnet"):
             cak['control_depth'] = depth_image