Spaces:
Runtime error
Runtime error
update
Browse files
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="
|
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"]
|