99eren99 commited on
Commit
5ef4b4e
·
verified ·
1 Parent(s): 64e160b

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -3
README.md CHANGED
@@ -39,13 +39,12 @@ tokenizer = AutoTokenizer.from_pretrained(
39
  #tokenizer.truncation_side = "right"
40
 
41
  model = AutoModelForMaskedLM.from_pretrained(
42
- "99eren99/ModernBERT-base-Turkish-uncased-mlm",
43
  )
44
 
45
  model.eval()
46
 
47
- # for moving to gpu
48
- # model.to("cuda", dtype=torch.float16)
49
 
50
  ```
51
 
 
39
  #tokenizer.truncation_side = "right"
40
 
41
  model = AutoModelForMaskedLM.from_pretrained(
42
+ "99eren99/ModernBERT-base-Turkish-uncased-mlm", torch_dtype="auto"
43
  )
44
 
45
  model.eval()
46
 
47
+ # model.to("cuda")
 
48
 
49
  ```
50