alexnasa commited on
Commit
a8c20ec
·
verified ·
1 Parent(s): 7bccef5

Update src/model/encoder/anysplat.py

Browse files
Files changed (1) hide show
  1. src/model/encoder/anysplat.py +1 -2
src/model/encoder/anysplat.py CHANGED
@@ -130,8 +130,7 @@ class EncoderAnySplat(Encoder[EncoderAnySplatCfg]):
130
 
131
  def __init__(self, cfg: EncoderAnySplatCfg) -> None:
132
  super().__init__(cfg)
133
- model_full = VGGT.from_pretrained("facebook/VGGT-1B")
134
- # model_full = VGGT()
135
  self.aggregator = model_full.aggregator.to(torch.bfloat16)
136
  self.freeze_backbone = cfg.freeze_backbone
137
  self.distill = cfg.distill
 
130
 
131
  def __init__(self, cfg: EncoderAnySplatCfg) -> None:
132
  super().__init__(cfg)
133
+ model_full = VGGT()
 
134
  self.aggregator = model_full.aggregator.to(torch.bfloat16)
135
  self.freeze_backbone = cfg.freeze_backbone
136
  self.distill = cfg.distill