Skip to content

Commit 750d2ab

Browse files
declare safetensors, average_checkpoints reads it
1 parent bb90318 commit 750d2ab

1 file changed

Lines changed: 1 addition & 2 deletions

File tree

etc/requirements-ml.txt

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -13,9 +13,8 @@ accelerate>=0.33
1313
# CRF layer on top of the token classifier
1414
pytorch-crf>=0.7.2
1515

16-
# reading the trained weights and pulling the model from the hub
16+
# reading and averaging the saved checkpoints
1717
safetensors>=0.4
18-
huggingface-hub>=0.24
1918

2019
# 8 bit optimizer to fit deberta-large on a 16gb gpu, optional at runtime
2120
bitsandbytes>=0.43

0 commit comments

Comments
 (0)