Skip to content

Commit 48bf710

Browse files
committed
Add size estimates for new models in setup mode
1 parent 2c987ee commit 48bf710

1 file changed

Lines changed: 6 additions & 0 deletions

File tree

src/setup.rs

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -31,6 +31,9 @@ fn model_size_gb(name: &str) -> Option<(f64, f64)> {
3131
"qwen3.5:27b" => Some((16.0, 18.0)),
3232
"qwen3.5:35b-a3b" => Some((20.0, 22.0)),
3333
"lfm2:24b-a2b" => Some((14.0, 16.0)),
34+
"qwen3.5:0.8b" => Some((0.5, 1.0)),
35+
"qwen3.5:2b" => Some((1.2, 2.0)),
36+
"nemotron:30b-a3b" => Some((24.5, 27.0)),
3437
_ => None,
3538
}
3639
}
@@ -449,6 +452,9 @@ mod tests {
449452
"qwen3.5:27b",
450453
"qwen3.5:35b-a3b",
451454
"lfm2:24b-a2b",
455+
"qwen3.5:0.8b",
456+
"qwen3.5:2b",
457+
"nemotron:30b-a3b",
452458
] {
453459
let (size, needed) = model_size_gb(name).unwrap();
454460
assert!(

0 commit comments

Comments
 (0)