Commit
·
db20d3e
1
Parent(s):
23e4ec1
limit chunk size
Browse files- app.py +1 -0
- tsr/system.py +1 -1
app.py
CHANGED
|
@@ -25,6 +25,7 @@ model = TSR.from_pretrained(
|
|
| 25 |
weight_name="model.ckpt",
|
| 26 |
token=HF_TOKEN
|
| 27 |
)
|
|
|
|
| 28 |
model.to(device)
|
| 29 |
|
| 30 |
rembg_session = rembg.new_session()
|
|
|
|
| 25 |
weight_name="model.ckpt",
|
| 26 |
token=HF_TOKEN
|
| 27 |
)
|
| 28 |
+
model.renderer.set_chunk_size(131072)
|
| 29 |
model.to(device)
|
| 30 |
|
| 31 |
rembg_session = rembg.new_session()
|
tsr/system.py
CHANGED
|
@@ -168,7 +168,7 @@ class TSR(BaseModule):
|
|
| 168 |
return
|
| 169 |
self.isosurface_helper = MarchingCubeHelper(resolution)
|
| 170 |
|
| 171 |
-
def extract_mesh(self, scene_codes, resolution: int = 256, threshold: float =
|
| 172 |
self.set_marching_cubes_resolution(resolution)
|
| 173 |
meshes = []
|
| 174 |
for scene_code in scene_codes:
|
|
|
|
| 168 |
return
|
| 169 |
self.isosurface_helper = MarchingCubeHelper(resolution)
|
| 170 |
|
| 171 |
+
def extract_mesh(self, scene_codes, resolution: int = 256, threshold: float = 25.0):
|
| 172 |
self.set_marching_cubes_resolution(resolution)
|
| 173 |
meshes = []
|
| 174 |
for scene_code in scene_codes:
|