File size: 672 Bytes
2d06dcc |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 |
from .bert import BERT, BERT_Norm, BERT_K_1_way, BERT_SEG, BERT_Disaware, BERT_DOC, BERT_MDF, BERT_MDF_Pretrain, BERT_KNNCL
from .llama import LLAMA_lora_Disaware
backbones_map = {
'bert': BERT,
'bert_norm': BERT_Norm,
'bert_K+1-way': BERT_K_1_way,
'bert_seg': BERT_SEG,
'bert_disaware': BERT_Disaware,
'bert_doc': BERT_DOC,
'bert_mdf': BERT_MDF,
'bert_mdf_pretrain': BERT_MDF_Pretrain,
'bert_knncl': BERT_KNNCL,
'llama_disaware': LLAMA_lora_Disaware,
}
|