Stable-X commited on
Commit
d2c4565
·
verified ·
1 Parent(s): 65e6388

Update trellis/pipelines/trellis_image_to_3d.py

Browse files
trellis/pipelines/trellis_image_to_3d.py CHANGED
@@ -336,7 +336,7 @@ class TrellisImageTo3DPipeline(Pipeline):
336
  """Lazy loading of the BiRefNet model"""
337
  from transformers import AutoImageProcessor, Mask2FormerForUniversalSegmentation, AutoModelForImageSegmentation
338
  self.birefnet_model = AutoModelForImageSegmentation.from_pretrained(
339
- 'weights/BiRefNet',
340
  trust_remote_code=True
341
  ).to(self.device)
342
  self.birefnet_model.eval()
 
336
  """Lazy loading of the BiRefNet model"""
337
  from transformers import AutoImageProcessor, Mask2FormerForUniversalSegmentation, AutoModelForImageSegmentation
338
  self.birefnet_model = AutoModelForImageSegmentation.from_pretrained(
339
+ 'ZhengPeng7/BiRefNet',
340
  trust_remote_code=True
341
  ).to(self.device)
342
  self.birefnet_model.eval()