Upload UncertaintyHead model and config

config.yaml

@@ -17,14 +17,14 @@ feature_extractor:
   - -15
   - -20
   - -25
-  attn_history_sz: 5
+  attn_history_sz: 2
   pool: false
 - name: luq.feature_extractors.token_probabilities
   top_n: 4
 uncertainty_head:
-  head_dim: 4096
-  n_layers: 1
-  n_heads: 32
-  dropout: 0.1
+  head_dim: 256
+  n_layers: 2
+  n_heads: 8
+  dropout: 0.2
   enable_feature_projection_layer: true
-  interim_dim: 2048
+  interim_dim: 256

weights.pth

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:015ea1c1aa52361e213affd6de4d41951d056132d5c3e81414d9235da95fde93
-size 611055840
+oid sha256:33e3ea4c4a7b536bbe359aa8ec953ba309aa1b51722c291db224a47bd988b15e
+size 44904609