Spaces:

Wuvin
/

Unique3D

Build error

Wuvin commited on May 31, 2024

Commit

4675486

1 Parent(s): 6babd58

try

Files changed (1) hide show

custum_3d_diffusion/custum_pipeline/unifield_pipeline_img2mvimg.py CHANGED Viewed

@@ -204,7 +204,8 @@ class StableDiffusionImage2MVCustomPipeline(
         #     batch_size = len(image)
         # else:
         #     batch_size = image.shape[0]
-        device = self._execution_device
         # here `guidance_scale` is defined analog to the guidance weight `w` of equation (2)
         # of the Imagen paper: https://arxiv.org/pdf/2205.11487.pdf . `guidance_scale = 1`
         # corresponds to doing no classifier free guidance.

         #     batch_size = len(image)
         # else:
         #     batch_size = image.shape[0]
+        # device = self._execution_device
+        device = "cuda"
         # here `guidance_scale` is defined analog to the guidance weight `w` of equation (2)
         # of the Imagen paper: https://arxiv.org/pdf/2205.11487.pdf . `guidance_scale = 1`
         # corresponds to doing no classifier free guidance.