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
|
@@ -819,9 +819,6 @@ class TrellisVGGTTo3DPipeline(TrellisImageTo3DPipeline):
|
|
| 819 |
new_pipeline.__dict__ = pipeline.__dict__
|
| 820 |
args = pipeline._pretrained_args
|
| 821 |
new_pipeline.VGGT_dtype = torch.float32
|
| 822 |
-
# VGGT_model = VGGT()
|
| 823 |
-
# VGGT_model_weight = torch.load("weights/VGGT_weight/object_vggt_model.pt", map_location=torch.device('cpu'))
|
| 824 |
-
# VGGT_model.load_state_dict(VGGT_model_weight)
|
| 825 |
VGGT_model = VGGT.from_pretrained("Stable-X/vggt-object-v0-1")
|
| 826 |
new_pipeline.VGGT_model = VGGT_model.to(new_pipeline.device)
|
| 827 |
del new_pipeline.VGGT_model.depth_head
|
|
|
|
| 819 |
new_pipeline.__dict__ = pipeline.__dict__
|
| 820 |
args = pipeline._pretrained_args
|
| 821 |
new_pipeline.VGGT_dtype = torch.float32
|
|
|
|
|
|
|
|
|
|
| 822 |
VGGT_model = VGGT.from_pretrained("Stable-X/vggt-object-v0-1")
|
| 823 |
new_pipeline.VGGT_model = VGGT_model.to(new_pipeline.device)
|
| 824 |
del new_pipeline.VGGT_model.depth_head
|