Learning/AntSK
1
0
Fork 0
mirror of https://github.com/AIDotNet/AntSK.git synced 2026-02-17 22:10:14 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix 调整llamafactory加载顺序

Browse Source
This commit is contained in:
zyxucp
2024-03-25 12:03:09 +08:00
parent 74e323158d
commit ced2a9b2e2
1 changed files with 9 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

17
src/AntSK/Pages/Setting/AIModel/AddModel.razor.cs
View File

@@ -67,6 +67,14 @@ namespace AntSK.Pages.Setting.AIModel
{
_aiModel = _aimodels_Repositories.GetFirst(p => p.Id == ModelId);
}
modelList = _ILLamaFactoryService.GetLLamaFactoryModels();
llamaFactoryDic = await _IDics_Repositories.GetFirstAsync(p => p.Type == LLamaFactoryConstantcs.LLamaFactorDic && p.Key == LLamaFactoryConstantcs.IsStartKey);
if (llamaFactoryDic != null)
{
llamaFactoryIsStart = llamaFactoryDic.Value == "false" ? false : true;
}
//目前只支持gguf的 所以筛选一下
_modelFiles = Directory.GetFiles(Path.Combine(Directory.GetCurrentDirectory(), LLamaSharpOption.FileDirectory)).Where(p=>p.Contains(".gguf")).ToArray();
if (!string.IsNullOrEmpty(ModelPath))
@@ -76,14 +84,7 @@ namespace AntSK.Pages.Setting.AIModel
_downloadModalVisible = true;
_downloadUrl = $"https://hf-mirror.com{ModelPath.Replace("---","/")}";
}
modelList = _ILLamaFactoryService.GetLLamaFactoryModels();
llamaFactoryDic = await _IDics_Repositories.GetFirstAsync(p => p.Type == LLamaFactoryConstantcs.LLamaFactorDic && p.Key == LLamaFactoryConstantcs.IsStartKey);
if (llamaFactoryDic != null)
{
llamaFactoryIsStart= llamaFactoryDic.Value== "false" ? false:true;
}
}
}
catch
{