Stable-X commited on
Commit
5bacbc7
·
verified ·
1 Parent(s): 6c326bb

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