Skip to content

Commit 521839e

Browse files
committed
refactor(ai): 优化模型下载脚本并调整显存计算逻辑
- 修复了模型列表中部分字段的缩进不一致问题 - 优化了模型内存和显存需求的计算逻辑 - 在获取 GPU 信息时,增加了最小显存限制(至少 8GB)
1 parent cfb7e8c commit 521839e

1 file changed

Lines changed: 22 additions & 20 deletions

File tree

ai/downloadModels.ps1

Lines changed: 22 additions & 20 deletions
Original file line numberDiff line numberDiff 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
# 显示系统信息

0 commit comments

Comments
 (0)