camie-tagger / config.yaml
Camais03's picture
Update config.yaml
1030e0f verified
model_info:
name: camie-tagger
version: 1.0.0
type: pytorch
models:
- name: initial
path: model/model_initial_only.pt
description: Initial version of the tagger model
- name: initial_st
path: model_initial.safetensors
description: Initial safetensors
- name: refined
path: model/model_refined.pt
description: Refined version of the tagger model with improved performance
- name: refined_st
path: model_refined.safetensors
description: Refined safetensors
- name: onnx
path: model_initial.onnx
description: Onnx model
default_model: initial
framework:
name: pytorch
version: "1.X"
metadata:
task: classification
tags:
- tagger
- classification
license: GPL-3.0