fixed the comment and showing params for models

studio/frontend/src/config/training.ts

@@ -148,7 +148,9 @@ export const PRIORITY_TRAINING_MODELS: readonly string[] = [
 
 /** Pin priority models to the top of a list of model IDs, preserving their defined order. */
 export function applyPriorityOrdering(ids: string[]): string[] {
-  const prioritySet = new Set<string>(PRIORITY_TRAINING_MODELS);
-  const rest = ids.filter((id) => !prioritySet.has(id));
-  return [...PRIORITY_TRAINING_MODELS, ...rest];
+  const idSet = new Set(ids);
+  const pinned = PRIORITY_TRAINING_MODELS.filter((id) => idSet.has(id));
+  const pinnedSet = new Set(pinned);
+  const rest = ids.filter((id) => !pinnedSet.has(id));
+  return [...pinned, ...rest];
 }

studio/frontend/src/features/onboarding/components/steps/model-selection-step.tsx

@@ -33,7 +33,7 @@ import {
   TooltipContent,
   TooltipTrigger,
 } from "@/components/ui/tooltip";
-import { MODEL_TYPE_TO_HF_TASK, applyPriorityOrdering } from "@/config/training";
+import { MODEL_TYPE_TO_HF_TASK, PRIORITY_TRAINING_MODELS, applyPriorityOrdering } from "@/config/training";
 import {
   useDebouncedValue,
   useGpuInfo,
@@ -96,6 +96,7 @@ export function ModelSelectionStep() {
     task,
     accessToken: hfToken || undefined,
     excludeGguf: true,
+    priorityIds: PRIORITY_TRAINING_MODELS,
   });
 
   const { error: tokenValidationError, isChecking: isCheckingToken } =

studio/frontend/src/features/studio/sections/model-section.tsx

@@ -28,7 +28,7 @@ import {
   TooltipContent,
   TooltipTrigger,
 } from "@/components/ui/tooltip";
-import { MODEL_TYPE_TO_HF_TASK, applyPriorityOrdering } from "@/config/training";
+import { MODEL_TYPE_TO_HF_TASK, PRIORITY_TRAINING_MODELS, applyPriorityOrdering } from "@/config/training";
 import {
   useDebouncedValue,
   useGpuInfo,
@@ -162,6 +162,7 @@ export function ModelSection() {
     task,
     accessToken: hfToken || undefined,
     excludeGguf: true,
+    priorityIds: PRIORITY_TRAINING_MODELS,
   });
 
   const { error: tokenValidationError, isChecking: isCheckingToken } =

studio/frontend/src/hooks/use-hf-model-search.ts

@@ -2,7 +2,7 @@
 // Copyright 2026-present the Unsloth AI Inc. team. All rights reserved. See /studio/LICENSE.AGPL-3.0
 
 import type { PipelineType } from "@huggingface/hub";
-import { listModels } from "@huggingface/hub";
+import { listModels, modelInfo } from "@huggingface/hub";
 import { useCallback, useMemo } from "react";
 import { useHfPaginatedSearch } from "./use-hf-paginated-search";
 
@@ -148,17 +148,71 @@ async function* mergedModelIterator(
   }
 }
 
+/**
+ * Creates an async generator that yields priority models (fetched individually
+ * via modelInfo for full metadata), then the general unsloth listing.
+ */
+async function* priorityThenListingIterator(
+  priorityIds: readonly string[],
+  task?: PipelineType,
+  accessToken?: string,
+): AsyncGenerator<unknown> {
+  const common = {
+    additionalFields: ["safetensors", "tags"] as ("safetensors" | "tags")[],
+    fetch: withPopularitySort,
+    ...(accessToken ? { credentials: { accessToken } } : {}),
+  };
+
+  // Phase 1: fetch priority models in parallel via modelInfo
+  const seen = new Set<string>();
+  const settled = await Promise.allSettled(
+    priorityIds.map((id) =>
+      modelInfo({
+        name: id,
+        additionalFields: ["safetensors", "tags"],
+        ...(accessToken ? { credentials: { accessToken } } : {}),
+      }),
+    ),
+  );
+  for (const result of settled) {
+    if (result.status === "fulfilled") {
+      const m = result.value as { name?: string };
+      if (m.name) seen.add(m.name);
+      yield result.value;
+    }
+  }
+
+  // Phase 2: yield general unsloth listing, skipping already-seen
+  const generalIter = listModels({
+    search: { owner: "unsloth", ...(task ? { task } : {}) },
+    ...common,
+  });
+  for await (const model of generalIter) {
+    const m = model as { name?: string };
+    if (m.name && seen.has(m.name)) continue;
+    yield model;
+  }
+}
+
 export function useHfModelSearch(
   query: string,
-  options?: { task?: PipelineType; accessToken?: string; excludeGguf?: boolean },
+  options?: {
+    task?: PipelineType;
+    accessToken?: string;
+    excludeGguf?: boolean;
+    priorityIds?: readonly string[];
+  },
 ) {
-  const { task, accessToken, excludeGguf = false } = options ?? {};
+  const { task, accessToken, excludeGguf = false, priorityIds } = options ?? {};
 
   const createIter = useCallback(
     () => {
       const trimmed = query.trim();
       if (!trimmed) {
-        // No query → show default unsloth models
+        // No query → show priority models first (with full metadata), then general unsloth listing
+        if (priorityIds && priorityIds.length > 0) {
+          return priorityThenListingIterator(priorityIds, task, accessToken) as AsyncGenerator<unknown>;
+        }
         return listModels({
           search: { owner: "unsloth", ...(task ? { task } : {}) },
           additionalFields: ["safetensors", "tags"],
@@ -169,7 +223,7 @@ export function useHfModelSearch(
       // Typed query: disable task filter so explicitly searched models still appear even if HF task metadata is wrong/missing.
       return mergedModelIterator(trimmed, undefined, accessToken) as AsyncGenerator<unknown>;
     },
-    [query, task, accessToken],
+    [query, task, accessToken, priorityIds],
   );
 
   const mapModel = useMemo(() => makeMapModel(excludeGguf), [excludeGguf]);