saplma_gemma-2-9b-it / config.yaml
Artem's picture
Update config.yaml
bf74220 verified
head_type: saplma
feature_extractor:
- name: luh.feature_extractors.basic_hidden_states
layer_nums: -1
uncertainty_head:
internal_dim1: 256
internal_dim2: 128
internal_dim3: 64