from trainer.distill import Distill,  DistillFFN, DistillProj
from trainer.finetune import FinetuneCLIP, FinetuneFFN, FinetuenProj, FinetuneTextProj, FinetuenProjTV
from trainer.frozenclip import FrozenCLIP
from trainer.nmc import FrozenNMC
from trainer.condsetft import CondFT
from trainer.conddistill import CondDistill,CondDistillFFN,CondDistillProj
from trainer.masedit import MASEDIT
from trainer.maskedit import MASKEDIT
from trainer.faredit import FAREDIT
from trainer.masklearn import MaskLearn
from trainer.mas import MAS
METHOD = {'FrozenCLIP': FrozenCLIP,

          'FrozenNMC': FrozenNMC,

          'Finetune': FinetuneCLIP,
          'finetunevisual': FinetuneCLIP,
          'FinetuneFFN': FinetuneFFN,
          'FinetuneCproj': FinetuenProj,
          'FinetuneCprojboth': FinetuenProjTV,
          'masedit': MASEDIT,
          'maskedit':MASKEDIT,
          'faredit': FAREDIT,
          'masklearn':MaskLearn,



          'FinetuneTextCproj': FinetuneTextProj,

          'distillboth': Distill,
          'distillvisual': Distill,
          'DistillFFN': DistillFFN,
          'DistillProj': DistillProj,


          'TextDistill': Distill,
          'TextDistillFFN': DistillFFN,
          'TextDistillProj': DistillProj,

          'CondFinetune':CondFT,
          'CondFinetunefirst':CondFT,
          'CondFinetuneVisual':CondFT,


          'CondDistill': CondDistill,
          'CondDistillFFN':CondDistillFFN,
          'CondDistillProj':CondDistillProj,
          'mas': MAS,



          }
