From 44b9d0d6766cb3412bedf5bea9730851deed7f1a Mon Sep 17 00:00:00 2001
From: kijai <40791699+kijai@users.noreply.github.com>
Date: Wed, 5 Feb 2025 17:18:27 +0200
Subject: [PATCH] Update vae.py

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

diff --git a/hy3dgen/shapegen/models/vae.py b/hy3dgen/shapegen/models/vae.py
index 16f88ca..b9b13bf 100755
--- a/hy3dgen/shapegen/models/vae.py
+++ b/hy3dgen/shapegen/models/vae.py
@@ -629,7 +629,7 @@ class ShapeVAE(nn.Module):
         xyz_samples = torch.FloatTensor(xyz_samples)
 
         if mc_algo == 'odc':
-            from ....occupancy_dual_contouring import occupancy_dual_contouring
+            from ....ODC.occupancy_dual_contouring import occupancy_dual_contouring
             odc = occupancy_dual_contouring(device=device)
 
         # 2. latents to 3d volume