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
|
@@ -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 |
-
'
|
| 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()
|