BBBAKERY's picture
Upload kcelectra/model.json with huggingface_hub
4507b2d verified
Raw
History Blame Contribute Delete
351 Bytes
{
"model": "beomi/KcELECTRA-base-v2022 + LoRA (merged)",
"num_classes": 7,
"labels": [
"happiness",
"anger",
"disgust",
"fear",
"neutral",
"sadness",
"surprise"
],
"input": "input_ids: (batch, seq_len) int64, attention_mask: (batch, seq_len) int64",
"output": "logits: (batch, 7) float32",
"max_length": 128
}