tim1900
/

bert-chunker

Token Classification

feature-extraction

Model card Files Files and versions

tim1900 commited on Dec 1, 2024

Commit

ffb9545

·

verified ·

1 Parent(s): 5002a78

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -39,7 +39,7 @@ config = AutoConfig.from_pretrained(
 # initialize model
 model = BertChunker(config)
-device='cuda' # or 'cpu'
 model.to(device)
 # load parameters, tim1900/BertChunker/model.safetensors

 # initialize model
 model = BertChunker(config)
+device='cpu' # or 'cuda'
 model.to(device)
 # load parameters, tim1900/BertChunker/model.safetensors