File tree Expand file tree Collapse file tree
Expand file tree Collapse file tree Original file line number Diff line number Diff line change @@ -20,32 +20,32 @@ Import-Module "$PSScriptRoot\..\psutils\index.psm1" -Force
2020# 模型列表,包含内存和显存需求信息(单位:GB)
2121$modelList = @ (
2222 @ {
23- ModelId = " bge-m3"
24- Name = ' bge-m3'
25- Size = 2
23+ ModelId = " bge-m3"
24+ Name = ' bge-m3'
25+ Size = 2
2626 VramRequired = 2
2727 },
2828 @ {
29- ModelId = " qwen3:8b"
30- Name = ' qwen3:8b'
31- Size = 8
29+ ModelId = " qwen3:8b"
30+ Name = ' qwen3:8b'
31+ Size = 8
3232 VramRequired = 8
3333 },
34- @ {
35- ModelId = " qwen3:14b"
36- Name = ' qwen3:14b'
34+ @ {
35+ ModelId = " qwen3:14b"
36+ Name = ' qwen3:14b'
3737 VramRequired = 14
3838 },
3939 @ {
40- ModelId = " gemma3:4b"
41- Name = ' gemma3:4b'
42- Size = 4
40+ ModelId = " gemma3:4b"
41+ Name = ' gemma3:4b'
42+ Size = 4
4343 VramRequired = 4
4444 },
4545 @ {
46- ModelId = " gemma3n:e4b"
47- Name = ' gemma3n:e4b'
48- Size = 4
46+ ModelId = " gemma3n:e4b"
47+ Name = ' gemma3n:e4b'
48+ Size = 4
4949 VramRequired = 4
5050 },
5151 # @{
@@ -55,9 +55,9 @@ $modelList = @(
5555 # VramRequired = 7
5656 # },
5757 @ {
58- ModelId = " qwen3:30b-a3b"
59- Name = ' qwen3:30b-a3b'
60- Size = 19
58+ ModelId = " qwen3:30b-a3b"
59+ Name = ' qwen3:30b-a3b'
60+ Size = 19
6161 VramRequired = 16
6262 }
6363)
@@ -89,8 +89,8 @@ function Test-ModelCanDownload {
8989 )
9090
9191
92- $modelMemoryGB = if ($Model.Size ){ $Model.Size } else { 8 }
93- $modelVramGB = if ($Model.VramRequired ){ $Model.VramRequired } else { 4 }
92+ $modelMemoryGB = if ($Model.Size ) { $Model.Size } else { 8 }
93+ $modelVramGB = if ($Model.VramRequired ) { $Model.VramRequired } else { 4 }
9494
9595 if ($GpuInfo.HasGpu ) {
9696 # 有GPU时检查显存
@@ -129,6 +129,8 @@ Write-Host "正在检测系统资源..." -ForegroundColor Yellow
129129
130130# 获取系统信息
131131$gpuInfo = Get-GpuInfo
132+ # 最小显存按照8gb
133+ $gpuInfo.VramGB = [math ]::max($gpuInfo.VramGB , 8 )
132134$memoryInfo = Get-SystemMemoryInfo
133135
134136# 显示系统信息
You can’t perform that action at this time.
0 commit comments