File size: 1,781 Bytes
3494c6b |
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 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 |
from ..models.rome import ROMEHyperParams, apply_rome_to_model
from ..models.memit import MEMITHyperParams, apply_memit_to_model
from ..models.kn import KNHyperParams, apply_kn_to_model
from ..models.mend import MENDHyperParams, MendRewriteExecutor, MendMultimodalRewriteExecutor
from ..models.ft import FTHyperParams, apply_ft_to_model
from ..models.serac import SERACHparams, SeracRewriteExecutor, SeracMultimodalRewriteExecutor
from ..dataset import ZsreDataset, CounterFactDataset, CaptionDataset, VQADataset
from ..models.ike import IKEHyperParams, apply_ike_to_model, apply_ike_to_multimodal_model
from ..models.ft_api import FTApiHyperParams, apply_ft_api_to_model
from ..models.lora import LoRAHyperParams, apply_lora_to_model
from ..models.grace import GraceHyperParams, apply_grace_to_model
from ..models.pmet import PMETHyperParams, apply_pmet_to_model
from ..models.melo import MELOHyperParams, apply_melo_to_model
ALG_DICT = {
'ROME': apply_rome_to_model,
'MEMIT': apply_memit_to_model,
"FT": apply_ft_to_model,
'KN': apply_kn_to_model,
'MEND': MendRewriteExecutor().apply_to_model,
'SERAC': SeracRewriteExecutor().apply_to_model,
'IKE': apply_ike_to_model,
'FT-Api': apply_ft_api_to_model,
'LoRA': apply_lora_to_model,
'GRACE': apply_grace_to_model,
'PMET': apply_pmet_to_model,
'MELO': apply_melo_to_model
}
ALG_MULTIMODAL_DICT = {
'MEND': MendMultimodalRewriteExecutor().apply_to_model,
'SERAC': SeracMultimodalRewriteExecutor().apply_to_model,
'SERAC_MULTI': SeracMultimodalRewriteExecutor().apply_to_model,
'IKE': apply_ike_to_multimodal_model,
}
DS_DICT = {
"cf": CounterFactDataset,
"zsre": ZsreDataset,
}
MULTIMODAL_DS_DICT = {
"caption": CaptionDataset,
"vqa": VQADataset,
}
|