bug fix

.gitignore

@@ -0,0 +1 @@
+*.pyc

mesh_reconstruction/render.py

@@ -12,8 +12,6 @@ def _warmup(glctx, device=None):
     tri = tensor([[0, 1, 2]], dtype=torch.int32)
     dr.rasterize(glctx, pos, tri, resolution=[256, 256])
 
-glctx = dr.RasterizeGLContext(output_db=False, device="cuda")
-
 class NormalsRenderer:
     
     _glctx:dr.RasterizeGLContext = None
@@ -31,7 +29,7 @@ class NormalsRenderer:
         else:
             self._mvp = mvp
         self._image_size = image_size
-        self._glctx = glctx
+        self._glctx = dr.RasterizeGLContext(output_db=False, device=device)
         _warmup(self._glctx, device)
 
     def render(self,

scripts/sd_model_zoo.py

@@ -65,7 +65,7 @@ def load_image_encoder():
     )
     return image_encoder
 
-def load_common_sd15_pipe(base_model=DEFAULT_BASE_MODEL, device="auto", controlnet=None, ip_adapter=False, plus_model=True, torch_dtype=torch.float16, model_cpu_offload_seq=None, enable_sequential_cpu_offload=False, vae_slicing=False, pipeline_class=None, **kwargs):
+def load_common_sd15_pipe(base_model=DEFAULT_BASE_MODEL, device="balanced", controlnet=None, ip_adapter=False, plus_model=True, torch_dtype=torch.float16, model_cpu_offload_seq=None, enable_sequential_cpu_offload=False, vae_slicing=False, pipeline_class=None, **kwargs):
     model_kwargs = dict(
         torch_dtype=torch_dtype, 
         device_map=device,
@@ -119,7 +119,7 @@ def load_common_sd15_pipe(base_model=DEFAULT_BASE_MODEL, device="auto", controln
     if enable_sequential_cpu_offload:
         pipe.enable_sequential_cpu_offload()
     else:
-        pipe = pipe.to("cuda")
+        # pipe = pipe.to("cuda")
         pass
         # pipe.enable_model_cpu_offload()
     if vae_slicing: