ikuinen99 commited on
Commit
d286d97
1 Parent(s): 957434c
Files changed (1) hide show
  1. imagebind/models/image_bind.py +1 -1
imagebind/models/image_bind.py CHANGED
@@ -643,7 +643,7 @@ class LayerNorm(nn.LayerNorm):
643
  return ret.type(orig_type)
644
 
645
 
646
- def load_ln_params(path="https://storage.googleapis.com/sfr-vision-language-research/LAVIS/models/BLIP2/blip2_pretrained_flant5xxl.pth"):
647
  state_dict = torch.load(path, map_location="cpu")["model"]
648
  params = type(state_dict)()
649
  params["weight"] = state_dict["ln_vision.weight"]
 
643
  return ret.type(orig_type)
644
 
645
 
646
+ def load_ln_params(path="checkpoints/blip2_pretrained_flant5xxl.pth"):
647
  state_dict = torch.load(path, map_location="cpu")["model"]
648
  params = type(state_dict)()
649
  params["weight"] = state_dict["ln_vision.weight"]