File size: 522 Bytes
6ca36e0 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 |
from backend.mistral7b import convert_with_mistral_7b
from backend.mistrallarge2 import convert_with_mistral_large2
from backend.codegen2 import convert_with_codegen2
def route_model_conversion(model_name, cs_code):
if "Mistral-7B" in model_name:
return convert_with_mistral_7b(cs_code)
elif "Mixtral" in model_name:
return convert_with_mistral_large2(cs_code)
elif "codegen2" in model_name:
return convert_with_codegen2(cs_code)
else:
return "Unsupported model selected."
|