MODELS_ID = dict(distilroberta = 'flax-sentence-embeddings/st-codesearch-distilroberta-base',
                 mpnet = 'flax-sentence-embeddings/all_datasets_v3_mpnet-base',
                 minilm_l6 = 'flax-sentence-embeddings/all_datasets_v3_MiniLM-L6')