Hugging Face's logo

llm-uncertainty-head
/
saplma_Mistral-7B-Instruct-v0.2

Model card Files Community
Artem commited on
Commit
061d71c
·
1 Parent(s): 1ef22bc

Upload UncertaintyHead model and config

Browse files
Files changed (2) hide show
  1. config.yaml +1 -1
  2. weights.pth +1 -1
config.yaml CHANGED
@@ -1,7 +1,7 @@
1
  head_type: saplma
2
  feature_extractor:
3
  - name: luq.feature_extractors.basic_hidden_states
4
- layer_nums: 32
5
  uncertainty_head:
6
  internal_dim1: 256
7
  internal_dim2: 128
 
1
  head_type: saplma
2
  feature_extractor:
3
  - name: luq.feature_extractors.basic_hidden_states
4
+ layer_nums: -1
5
  uncertainty_head:
6
  internal_dim1: 256
7
  internal_dim2: 128
weights.pth CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:fdcb34a6ab164528275e35a70cf35ba9d761b570fd64edd87182b5da81563ea8
3
  size 4363008
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:f8b3b7ca9cb5c1570a7becd451b69e146ad8862b1adf6e117bb6df2425fe9747
3
  size 4363008