Update preprocessor_normaldsine.py

lllyasviel · Mar 4, 2024 · 85f4206 · 85f4206
1 parent 4cf5bce
commit 85f4206
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/extensions-builtin/forge_preprocessor_normaldsine/scripts/preprocessor_normaldsine.py b/extensions-builtin/forge_preprocessor_normaldsine/scripts/preprocessor_normaldsine.py
@@ -65,7 +65,7 @@ def __call__(self, input_image, resolution, slider_1=None, slider_2=None, slider
             image_normal = rearrange(image_normal, 'h w c -> 1 c h w')
             image_normal = self.norm(image_normal)
 
-            intrins = get_intrins_from_fov(new_fov=fov, H=orig_H, W=orig_W, device=self.device).unsqueeze(0)
+            intrins = get_intrins_from_fov(new_fov=fov, H=orig_H, W=orig_W, device=torch.device('cpu')).unsqueeze(0)
             intrins[:, 0, 2] += l
             intrins[:, 1, 2] += t