myrkur
/

Persian-ModernBert-base

flash-attention

Inference Endpoints

Model card Files Files and versions Community

myrkur commited on Jan 4

Commit

f05dd97

·

verified ·

1 Parent(s): 87c5c2d

Update README.md

Files changed (1) hide show

README.md +2 -2

README.md CHANGED Viewed

@@ -58,7 +58,7 @@ tokenizer = AutoTokenizer.from_pretrained("myrkur/Persian-ModernBert-base")
 model = AutoModelForMaskedLM.from_pretrained("myrkur/Persian-ModernBert-base", attn_implementation="eager", torch_dtype=torch.bfloat16, device_map="cpu")
 ```
-### Example: Masked Token Prediction
 ```python
 text = "حال و [MASK] مردم خوب است."
@@ -88,7 +88,7 @@ tokenizer = AutoTokenizer.from_pretrained("myrkur/Persian-ModernBert-base")
 model = AutoModelForMaskedLM.from_pretrained("myrkur/Persian-ModernBert-base", attn_implementation="flash_attention_2", torch_dtype=torch.bfloat16, device_map="cuda")
 ```
-### Example: Masked Token Prediction
 ```python
 text = "حال و [MASK] مردم خوب است."

 model = AutoModelForMaskedLM.from_pretrained("myrkur/Persian-ModernBert-base", attn_implementation="eager", torch_dtype=torch.bfloat16, device_map="cpu")
 ```
+#### Example: Masked Token Prediction
 ```python
 text = "حال و [MASK] مردم خوب است."
 model = AutoModelForMaskedLM.from_pretrained("myrkur/Persian-ModernBert-base", attn_implementation="flash_attention_2", torch_dtype=torch.bfloat16, device_map="cuda")
 ```
+#### Example: Masked Token Prediction
 ```python
 text = "حال و [MASK] مردم خوب است."