tim1900
/

bert-chunker

Token Classification

feature-extraction

Model card Files Files and versions

tim1900 commited on Jul 6, 2024

Commit

6b26f41

·

verified ·

1 Parent(s): c478ac9

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -45,7 +45,7 @@ device='cuda'
 model.to(device)
 # load parameters
-state_dict = safetensors.torch.load_file("outputModels/checkpoint-3750/model.safetensors")
 model.load_state_dict(state_dict)
 # text to be chunked

 model.to(device)
 # load parameters
+state_dict = safetensors.torch.load_file("./model.safetensors")
 model.load_state_dict(state_dict)
 # text to be chunked