From d618646a888b2554aa7517c2585a84cbaba56c39 Mon Sep 17 00:00:00 2001
From: kijai <40791699+kijai@users.noreply.github.com>
Date: Mon, 3 Feb 2025 19:49:18 +0200
Subject: [PATCH] Update conditioner.py

---
 hy3dgen/shapegen/models/conditioner.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/hy3dgen/shapegen/models/conditioner.py b/hy3dgen/shapegen/models/conditioner.py
index e7841ca..3616fca 100755
--- a/hy3dgen/shapegen/models/conditioner.py
+++ b/hy3dgen/shapegen/models/conditioner.py
@@ -78,7 +78,7 @@ class ImageEncoder(nn.Module):
             mask = mask.to(image)
             image = image * mask
 
-        inputs = image
+        inputs = self.transform(image)
         outputs = self.model(inputs)
 
         last_hidden_state = outputs.last_hidden_state