THU-IAR's picture
Upload 198 files
2d06dcc verified
raw
history blame contribute delete
672 Bytes
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,
}