Update README.md

README.md

@@ -9,8 +9,6 @@ tags:
 
 **This is repository for MutBERT (pretrained with mutation data in human genome)**.
 
-**You can find all MutBERT variants at [here](https://huggingface.co/JadenLong).**
-
 ## Introduction
 
 This is the official pre-trained model introduced in MutBERT: Probabilistic Genome Representation Improves Genomics Foundation Models.
@@ -31,8 +29,8 @@ MutBERT is a transformer-based genome foundation model trained only on Human gen
 ```python
 from transformers import AutoTokenizer, AutoModel
 
-model_name = "JadenLong/MutBERT"
-# Optional: JadenLong/MutBERT-Huamn-Ref, JadenLong/MutBERT-Multi
+model_name = "CompBioDSA/MutBERT"
+# Optional: CompBioDSA/MutBERT-Huamn-Ref, CompBioDSA/MutBERT-Multi
 tokenizer = AutoTokenizer.from_pretrained(model_name)
 model = AutoModel.from_pretrained(model_name, trust_remote_code=True)
 ```
@@ -47,8 +45,8 @@ import torch.nn.functional as F
 
 from transformers import AutoTokenizer, AutoModel
 
-model_name = "JadenLong/MutBERT"
-# Optional: JadenLong/MutBERT-Huamn-Ref, JadenLong/MutBERT-Multi
+model_name = "CompBioDSA/MutBERT"
+# Optional: CompBioDSA/MutBERT-Huamn-Ref, CompBioDSA/MutBERT-Multi
 tokenizer = AutoTokenizer.from_pretrained(model_name)
 model = AutoModel.from_pretrained(model_name, trust_remote_code=True)
 
@@ -73,8 +71,8 @@ print(embedding_max.shape) # expect to be 768
 ```python
 from transformers import AutoModelForSequenceClassification
 
-model_name = "JadenLong/MutBERT"
-# Optional: JadenLong/MutBERT-Huamn-Ref, JadenLong/MutBERT-Multi
+model_name = "CompBioDSA/MutBERT"
+# Optional: CompBioDSA/MutBERT-Huamn-Ref, CompBioDSA/MutBERT-Multi
 model = AutoModelForSequenceClassification.from_pretrained(model_name, trust_remote_code=True, num_labels=2)
 ```
 
@@ -85,8 +83,8 @@ Allowed types for RoPE scaling are: `linear` and `dynamic`. To extend the model'
 If you want to scale your model context by 2x:
 
 ```python
-model_name = "JadenLong/MutBERT"
-# Optional: JadenLong/MutBERT-Huamn-Ref, JadenLong/MutBERT-Multi
+model_name = "CompBioDSA/MutBERT"
+# Optional: CompBioDSA/MutBERT-Huamn-Ref, CompBioDSA/MutBERT-Multi
 model = AutoModel.from_pretrained(model_name,
                                   trust_remote_code=True,
                                   rope_scaling={'type': 'dynamic','factor': 2.0}