- Update BART settings

rockwalrus · Jan 9, 2022 · b62781e · b62781e
1 parent 1bc326d
commit b62781e
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/ecco/model-config.yaml b/src/ecco/model-config.yaml
@@ -204,10 +204,10 @@ bigscience/T0pp:
  partial_token_prefix: ''
 # BART
 facebook/bart-large-mnli:
- embedding: 'shared'
+ embedding: 'model.shared'
  type: 'enc-dec'
  activations:
- - 'fc2'
+ - 'fc1'
  token_prefix: ''
  partial_token_prefix: '##'