microsoft
/

SatCLIP-ViT16-L10

Model card Files Files and versions

davanstrien HF Staff commited on Jan 16, 2024

Commit

0382586

·

verified ·

1 Parent(s): 7ddc6f0

Create README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -45,7 +45,7 @@ device = "cuda" if torch.cuda.is_available() else "cpu"
 c = torch.randn(32, 2)  # Represents a batch of 32 locations (lon/lat)
 model = get_satclip(
-    hf_hub_download("microsoft/SatCLIP-ViT16-L10", "satclip-vit16-l40.ckpt"),
     device=device,
 )  # Only loads location encoder by default
 model.eval()

 c = torch.randn(32, 2)  # Represents a batch of 32 locations (lon/lat)
 model = get_satclip(
+    hf_hub_download("microsoft/SatCLIP-ViT16-L10", "satclip-vit16-l10.ckpt"),
     device=device,
 )  # Only loads location encoder by default
 model.eval()