raphaelsty
/

splade-max

Model card Files Files and versions

raphaelsty commited on Aug 27, 2023

Commit

b2b7351

·

1 Parent(s): 2e6cde3

Update README.md

Files changed (1) hide show

README.md +2 -2

README.md CHANGED Viewed

@@ -35,8 +35,8 @@ documents = [
 }]
 model = model.Splade(
-    model=AutoModelForMaskedLM.from_pretrained("raphaelsty/splade_max").to(device),
-    tokenizer=AutoTokenizer.from_pretrained("raphaelsty/splade_max"),
     device=device
 )

 }]
 model = model.Splade(
+    model=AutoModelForMaskedLM.from_pretrained("raphaelsty/splade-max").to(device),
+    tokenizer=AutoTokenizer.from_pretrained("raphaelsty/splade-max"),
     device=device
 )