Spaces:
Running
on
Zero
Running
on
Zero
Update trellis/pipelines/trellis_image_to_3d.py
Browse files
trellis/pipelines/trellis_image_to_3d.py
CHANGED
@@ -230,11 +230,7 @@ class TrellisImageTo3DPipeline(Pipeline):
|
|
230 |
scale = min(1, 1024 / max_size)
|
231 |
if scale < 1:
|
232 |
input = input.resize((int(input.width * scale), int(input.height * scale)), Image.Resampling.LANCZOS)
|
233 |
-
|
234 |
-
# Load BiRefNet model if not already loaded
|
235 |
-
if getattr(self, 'birefnet_model', None) is None:
|
236 |
-
self._lazy_load_birefnet()
|
237 |
-
|
238 |
# Get mask using BiRefNet
|
239 |
mask = self._get_birefnet_mask(input)
|
240 |
|
|
|
230 |
scale = min(1, 1024 / max_size)
|
231 |
if scale < 1:
|
232 |
input = input.resize((int(input.width * scale), int(input.height * scale)), Image.Resampling.LANCZOS)
|
233 |
+
|
|
|
|
|
|
|
|
|
234 |
# Get mask using BiRefNet
|
235 |
mask = self._get_birefnet_mask(input)
|
236 |
|