nielsr HF Staff commited on
Commit
efc2ff9
·
verified ·
1 Parent(s): 453d493

Remove library name

Browse files

It doesn't seem like the model is compatible with the Transformers library.

Btw, you could consider making your `CopernicusFMViT` class inherit from the [PyTorchModelHubMixin](https://huggingface.co/docs/huggingface_hub/package_reference/mixins#huggingface_hub.PyTorchModelHubMixin) class. This way, you can do:

```python
from src.model_vit import CopernicusFMViT

# load model
model = CopernicusFMViT(...)

# equip with weights
filepath = hf_hub_download(repo_id="wangyi111/Copernicus-FM", filename="")
state_dict = torch.load(filepath, map_location="cpu")
model.load_state_dict(state_dict)

# push to the hub
model.push_to_hub("wangyi111/Copernicus-FM")

# now anyone can use it like so
model = CopernicusFMViT.from_pretrained("wangyi111/Copernicus-FM")
```

This comes with a few benefits:
- download metrics
- safetensors for weights serialization

Files changed (1) hide show
  1. README.md +0 -1
README.md CHANGED
@@ -9,7 +9,6 @@ tags:
9
  - remote-sensing
10
  - sentinel
11
  - foundation-model
12
- library_name: transformers
13
  ---
14
 
15
  # Model Card for Copernicus-FM
 
9
  - remote-sensing
10
  - sentinel
11
  - foundation-model
 
12
  ---
13
 
14
  # Model Card for Copernicus-FM