Stable-X commited on
Commit
984a9c4
·
verified ·
1 Parent(s): 3d094b5

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