From 24261cc59dd2c77b3b0d8319b5221116942c18df Mon Sep 17 00:00:00 2001
From: smthemex <138738845+smthemex@users.noreply.github.com>
Date: Sat, 21 Jun 2025 09:04:46 +0800
Subject: [PATCH] Update model.py

<10G ram
---
 PartPacker/vae/model.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/PartPacker/vae/model.py b/PartPacker/vae/model.py
index cc4598d..87d3580 100644
--- a/PartPacker/vae/model.py
+++ b/PartPacker/vae/model.py
@@ -332,7 +332,7 @@ class Model(nn.Module):
         self,
         data: dict[str, torch.Tensor],
         mode: Literal["dense", "hierarchical"] = "hierarchical",
-        max_samples_per_iter: int = 512**2,
+        max_samples_per_iter: int =8192,# 512**2 #for low ram <10G
         resolution: int = 512,
         min_resolution: int = 64,  # for hierarchical
     ) -> dict[str, torch.Tensor]: