ceos/unsloth
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Merge pull request #211 from unslothai/fix/param-count-display

Browse source 
Fix: Remove download count fallback when model param count is unavailable
This commit is contained in:
Roland Tannous 2026-02-22 16:52:25 +04:00 • committed by GitHub
commit a490dca8a0
3 changed files with 5 additions and 20 deletions
Download patch file Download diff file Expand all files Collapse all files

15
studio/frontend/src/components/assistant-ui/model-selector/pickers.tsx
View file

@ -151,12 +151,9 @@ export function HubModelPicker({
const metricsById = useMemo(
() =>
new Map(
results.map((result) => [
result.id,
result.totalParams
? formatCompact(result.totalParams)
: `↓${formatCompact(result.downloads)}`,
]),
results
.filter((result) => result.totalParams)
.map((result) => [result.id, formatCompact(result.totalParams!)]),
),
[results],
);
@ -167,11 +164,7 @@ export function HubModelPicker({
{ est: number; status: VramFitStatus | null; detail: string | null }
>();
for (const r of results) {
const detail = r.totalParams
? formatCompact(r.totalParams)
: r.downloads != null
? `↓${formatCompact(r.downloads)}`
: null;
const detail = r.totalParams ? formatCompact(r.totalParams) : null;
if (r.totalParams) {
const est = estimateLoadingVram(r.totalParams, "qlora");
const status = gpu.available

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

@ -244,10 +244,6 @@ export function ModelSelectionStep() {
<span className="text-xs text-muted-foreground shrink-0">
{sizeLabel}
</span>
) : r?.downloads != null ? (
<span className="text-[10px] text-muted-foreground shrink-0">
↓{formatCompact(r.downloads)}
</span>
) : null}
</ComboboxItem>
);

6
studio/frontend/src/features/studio/sections/model-section.tsx
View file

@ -210,11 +210,7 @@ export function ModelSection() {
{ est: number; status: VramFitStatus | null; detail: string | null }
>();
for (const r of hfResults) {
const detail = r.totalParams
? formatCompact(r.totalParams)
: r.downloads != null
? `\u2193${formatCompact(r.downloads)}`
: null;
const detail = r.totalParams ? formatCompact(r.totalParams) : null;
if (r.totalParams) {
const est = estimateLoadingVram(r.totalParams, method);
const status = gpu.available