diff --git a/app/components/model-select.tsx b/app/components/model-select.tsx index 4c4caf6d..5cd71d35 100644 --- a/app/components/model-select.tsx +++ b/app/components/model-select.tsx @@ -207,11 +207,6 @@ const ModelSelect: React.FC = ({ return counts; }; - const modelCounts = countModelsPerFamily(availableModels); - const sortedModelFamilies = Object.entries(modelFamilies).sort( - ([a], [b]) => (modelCounts[b] || 0) - (modelCounts[a] || 0), - ); - return (
@@ -222,20 +217,22 @@ const ModelSelect: React.FC = ({ inputRef={searchInputRef} />
- {sortedModelFamilies.map(([key, { name, icon: Icon }]) => ( - handleToggleFamilyFilter(key)} - bordered - text={name} - icon={ -
- {Icon ? : } -
- } - className={`${style["model-family-button"]}${selectedFamilies.includes(key) ? " " + style["selected-model-family"] : ""}`} - /> - ))} + {Object.entries(modelFamilies).map( + ([key, { name, icon: Icon }]) => ( + handleToggleFamilyFilter(key)} + bordered + text={name} + icon={ +
+ {Icon ? : } +
+ } + className={`${style["model-family-button"]}${selectedFamilies.includes(key) ? " " + style["selected-model-family"] : ""}`} + /> + ), + )}
diff --git a/app/utils/model.tsx b/app/utils/model.tsx index 5623c388..9dae4fa9 100644 --- a/app/utils/model.tsx +++ b/app/utils/model.tsx @@ -73,20 +73,20 @@ export interface ModelDetails { export const modelDetailsList: ModelDetails[] = [ { family: ModelFamily.LLAMA, name: "Llama", icon: MetaIcon }, { family: ModelFamily.DEEPSEEK, name: "DeepSeek", icon: DeepSeekIcon }, - { family: ModelFamily.PHI, name: "Phi", icon: MicrosoftIcon }, - { family: ModelFamily.MISTRAL, name: "Mistral", icon: MistralIcon }, - { family: ModelFamily.GEMMA, name: "Gemma", icon: GoogleIcon }, { family: ModelFamily.QWEN, name: "Qwen", icon: (...props) => Qwen Logo, }, + { family: ModelFamily.GEMMA, name: "Gemma", icon: GoogleIcon }, + { family: ModelFamily.PHI, name: "Phi", icon: MicrosoftIcon }, + { family: ModelFamily.MISTRAL, name: "Mistral", icon: MistralIcon }, { family: ModelFamily.SMOL_LM, name: "SmolLM", icon: (...props) => SmolLM Logo, }, - { family: ModelFamily.WIZARD_MATH, name: "Wizard Math", icon: WandSparkles }, { family: ModelFamily.STABLE_LM, name: "StableLM", icon: StablelmIcon }, { family: ModelFamily.REDPAJAMA, name: "RedPajama", icon: Shirt }, + { family: ModelFamily.WIZARD_MATH, name: "Wizard Math", icon: WandSparkles }, ];