weicaijaden commited on
Commit
a6cdc94
·
verified ·
1 Parent(s): cd005b4

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -35,7 +35,7 @@ tokenizer = AutoTokenizer.from_pretrained(model_name)
35
  model = AutoModel.from_pretrained(model_name, trust_remote_code=True)
36
  ```
37
 
38
- The default attention is flash attention("sdpa"). If you want use basic attention, you can replace it with "eager". Please refer to [here](https://huggingface.co/JadenLong/MutBERT/blob/main/modeling_mutbert.py#L438).
39
 
40
  ### Get embeddings
41
 
 
35
  model = AutoModel.from_pretrained(model_name, trust_remote_code=True)
36
  ```
37
 
38
+ The default attention is flash attention("sdpa"). If you want use basic attention, you can replace it with "eager". Please refer to [here](https://huggingface.co/CompBioDSA/MutBERT/blob/main/modeling_mutbert.py#L438).
39
 
40
  ### Get embeddings
41