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 |
|