Coverage for mindsdb / integrations / handlers / huggingface_handler / settings.py: 0%

1from mindsdb.integrations.handlers.huggingface_handler.finetune import ( 

2 _finetune_cls, 

3 _finetune_fill_mask, 

4 _finetune_question_answering, 

5 _finetune_summarization, 

6 _finetune_text_generation, 

7 _finetune_translate, 

8) 

9 

10# todo once we have moved predict tasks functions into a separate function 

11# PREDICT_MAP = { 

12# 'text-classification': self.predict_text_classification, 

13# 'zero-shot-classification': self.predict_zero_shot, 

14# 'translation': self.predict_translation, 

15# 'summarization': self.predict_summarization, 

16# 'fill-mask': self.predict_fill_mask 

17# } 

18 

19FINETUNE_MAP = { 

20 "text-classification": _finetune_cls, 

21 "zero-shot-classification": _finetune_cls, 

22 "translation": _finetune_translate, 

23 "summarization": _finetune_summarization, 

24 "fill-mask": _finetune_fill_mask, 

25 "text-generation": _finetune_text_generation, 

26 "question-answering": _finetune_question_answering, 

27}