From 7e234a017e9191b97f0a5096b7eafaeac54791fe Mon Sep 17 00:00:00 2001
From: kijai <40791699+kijai@users.noreply.github.com>
Date: Sat, 25 Jan 2025 23:32:24 +0200
Subject: [PATCH] make mask do something

---
 hy3dgen/shapegen/models/conditioner.py | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/hy3dgen/shapegen/models/conditioner.py b/hy3dgen/shapegen/models/conditioner.py
index 1af4c0c..3616fca 100755
--- a/hy3dgen/shapegen/models/conditioner.py
+++ b/hy3dgen/shapegen/models/conditioner.py
@@ -73,6 +73,11 @@ class ImageEncoder(nn.Module):
             image = (image - low) / (high - low)
 
         image = image.to(self.model.device, dtype=self.model.dtype)
+        
+        if mask is not None:
+            mask = mask.to(image)
+            image = image * mask
+
         inputs = self.transform(image)
         outputs = self.model(inputs)