Update config.yaml
Browse files- config.yaml +8 -0
config.yaml
CHANGED
@@ -7,11 +7,19 @@ models:
|
|
7 |
- name: initial
|
8 |
path: model/model_initial_only.pt
|
9 |
description: Initial version of the tagger model
|
|
|
|
|
|
|
|
|
10 |
|
11 |
- name: refined
|
12 |
path: model/model_refined.pt
|
13 |
description: Refined version of the tagger model with improved performance
|
14 |
|
|
|
|
|
|
|
|
|
15 |
- name: onnx
|
16 |
path: model_initial.onnx
|
17 |
description: Onnx model
|
|
|
7 |
- name: initial
|
8 |
path: model/model_initial_only.pt
|
9 |
description: Initial version of the tagger model
|
10 |
+
|
11 |
+
- name: initial_st
|
12 |
+
path: model_initial.safetensors
|
13 |
+
description: Initial safetensors
|
14 |
|
15 |
- name: refined
|
16 |
path: model/model_refined.pt
|
17 |
description: Refined version of the tagger model with improved performance
|
18 |
|
19 |
+
- name: refined_st
|
20 |
+
path: model_refined.safetensors
|
21 |
+
description: Refined safetensors
|
22 |
+
|
23 |
- name: onnx
|
24 |
path: model_initial.onnx
|
25 |
description: Onnx model
|