Hugging Face's logo

Spaces:
Tzktz
/
Dit-document-layout-analysis
Sleeping

App Files Community
Dit-document-layout-analysis
File size: 768 Bytes 
6fc683c
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
 
MODEL_NAME_TO_POOL_TYPE = {
    'e5-small': 'avg',
    'e5-base': 'avg',
    'e5-large': 'avg',
    'e5-small-v2': 'avg',
    'e5-base-v2': 'avg',
    'e5-large-v2': 'avg',
    'multilingual-e5-small': 'avg',
    'multilingual-e5-base': 'avg',
    'multilingual-e5-large': 'avg',
    'e5-mistral-7b-instruct': 'last',
}


MODEL_NAME_TO_PREFIX_TYPE = {
    'e5-small': 'query_or_passage',
    'e5-base': 'query_or_passage',
    'e5-large': 'query_or_passage',
    'e5-small-v2': 'query_or_passage',
    'e5-base-v2': 'query_or_passage',
    'e5-large-v2': 'query_or_passage',
    'multilingual-e5-small': 'query_or_passage',
    'multilingual-e5-base': 'query_or_passage',
    'multilingual-e5-large': 'query_or_passage',
    'e5-mistral-7b-instruct': 'instruction',
}