| 9 | # YiPPLBasedOnProbabilityTransferLogitsPIQAProcessor, \ |
| 10 | |
| 11 | class ModelProcessorFactory(): |
| 12 | @staticmethod |
| 13 | def create_processor(processor_type, **kwargs): |
| 14 | processor_classes = { |
| 15 | |
| 16 | "based_on_probility_transfer_logits_fp32_processor": BasedOnProbabilityTransferLogits_Loacal_FP32_Processor, |
| 17 | "based_on_probility_transfer_logits_fp32_digit_vote_processor": BasedOnProbabilityTransferLogits_Loacal_FP32_digit_vote_Processor, |
| 18 | |
| 19 | "yi_ppl_based_on_probility_transfer_logits_processor": YiPPLBasedOnProbabilityTransferLogitsProcessor, |
| 20 | "intermlm_ppl_based_on_probility_transfer_logits_processor": InternLMPPLBasedOnProbabilityTransferLogitsProcessor, |
| 21 | |
| 22 | # "yi_ppl_based_on_probility_transfer_logits_piqa_processor": YiPPLBasedOnProbabilityTransferLogitsPIQAProcessor |
| 23 | |
| 24 | } |
| 25 | selected_processor_class = processor_classes.get(processor_type.lower()) |
| 26 | if selected_processor_class: |
| 27 | return selected_processor_class(**kwargs) |
| 28 | else: |
| 29 | print(f"Unsupported processor type: {processor_type}") |
| 30 | return None |