seslami-pplx commited on
Commit
7cadae0
·
1 Parent(s): 05343bd

fix: update config for TEI

Browse files
Files changed (2) hide show
  1. README.md +1 -1
  2. config.json +2 -1
README.md CHANGED
@@ -71,7 +71,7 @@ curl -X POST https://api.perplexity.ai/v1/contextualizedembeddings \
71
  from transformers import AutoModel
72
 
73
  model_ctx = AutoModel.from_pretrained(
74
- "perplexity-ai/pplx-embed-1-context-0.6B",
75
  trust_remote_code=True
76
  )
77
 
 
71
  from transformers import AutoModel
72
 
73
  model_ctx = AutoModel.from_pretrained(
74
+ "perplexity-ai/pplx-embed-1-context-4B",
75
  trust_remote_code=True
76
  )
77
 
config.json CHANGED
@@ -72,5 +72,6 @@
72
  "use_cache": false,
73
  "use_sliding_window": false,
74
  "attn_implementation": "sdpa",
75
- "vocab_size": 151936
 
76
  }
 
72
  "use_cache": false,
73
  "use_sliding_window": false,
74
  "attn_implementation": "sdpa",
75
+ "vocab_size": 151936,
76
+ "use_bidirectional_attention": true
77
  }