fix: resolve boundary index calculation error in 3D mesh generation and AttributeError in image generator cleanup

models/image_generator.py

@@ -271,7 +271,7 @@ class OmniGenImageGenerator:
     
     def __del__(self):
         """Cleanup when object is destroyed"""
-        if self.pipeline:
+        if hasattr(self, 'pipeline') and self.pipeline is not None:
             del self.pipeline
         gc.collect()
         if torch.cuda.is_available():

models/model_3d_generator.py

@@ -335,21 +335,40 @@ class Hunyuan3DGenerator:
         base_vertices = []
         base_faces = []
         
-        # Get boundary vertices
+        # Get boundary vertices - fix the indexing
         boundary_indices = []
-        for i in range(h):
-            boundary_indices.append(i * w)  # Left edge
-            boundary_indices.append(i * w + w - 1)  # Right edge
+        
+        # Top edge (excluding corners)
         for j in range(1, w-1):
-            boundary_indices.append(j)  # Top edge
-            boundary_indices.append((h-1) * w + j)  # Bottom edge
+            boundary_indices.append(j)
+        
+        # Right edge (including top-right corner)
+        for i in range(h):
+            boundary_indices.append(i * w + w - 1)
+        
+        # Bottom edge (excluding bottom-right corner, going right to left)
+        for j in range(w-2, 0, -1):
+            boundary_indices.append((h-1) * w + j)
+        
+        # Left edge (including bottom-left corner, going bottom to top)
+        for i in range(h-1, -1, -1):
+            boundary_indices.append(i * w)
+        
+        # Remove duplicate indices (first and last should not be the same)
+        if boundary_indices and boundary_indices[0] == boundary_indices[-1]:
+            boundary_indices = boundary_indices[:-1]
         
         # Create base vertices
         start_idx = len(vertices)
         for idx in boundary_indices:
-            v = vertices[idx].copy()
-            v[2] = base_z
-            base_vertices.append(v)
+            if idx < len(vertices):  # Safety check
+                v = vertices[idx].copy()
+                v[2] = base_z
+                base_vertices.append(v)
+        
+        if not base_vertices:
+            # If no base vertices were created, return empty arrays
+            return np.array([]), np.array([])
         
         # Create center vertex
         center = np.mean(base_vertices, axis=0)