Hugging Face's logo

wallacelw
/
ModBERTBr

Safetensors
modernbert
Model card Files
xet
Community
wallacelw commited on
Commit
8fc9d35
·
verified ·
1 Parent(s): d800382

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -2
README.md CHANGED
@@ -29,7 +29,7 @@ Using `AutoModelForMaskedLM`:
29
 
30
  ```python
31
  from transformers import AutoTokenizer, AutoModelForMaskedLM
32
- model_id = "answerdotai/ModernBERT-base"
33
  tokenizer = AutoTokenizer.from_pretrained(model_id)
34
  model = AutoModelForMaskedLM.from_pretrained(model_id)
35
  text = "The capital of France is [MASK]."
@@ -52,7 +52,6 @@ from pprint import pprint
52
  pipe = pipeline(
53
  "fill-mask",
54
  model="answerdotai/ModernBERT-base",
55
- torch_dtype=torch.bfloat16,
56
  )
57
  input_text = "He walked to the [MASK]."
58
  results = pipe(input_text)
 
29
 
30
  ```python
31
  from transformers import AutoTokenizer, AutoModelForMaskedLM
32
+ model_id = "wallacelw/ModBERTBr"
33
  tokenizer = AutoTokenizer.from_pretrained(model_id)
34
  model = AutoModelForMaskedLM.from_pretrained(model_id)
35
  text = "The capital of France is [MASK]."
 
52
  pipe = pipeline(
53
  "fill-mask",
54
  model="answerdotai/ModernBERT-base",
 
55
  )
56
  input_text = "He walked to the [MASK]."
57
  results = pipe(input_text)