Mirrors/flowframes
1
0
Fork 0
mirror of https://github.com/n00mkrad/flowframes.git synced 2025-12-24 12:19:27 +01:00
Code Issues Packages Projects Releases Wiki Activity

Added support for custom models

Browse Source 
Dir name must end with "_custom", must be only A-Z, 0-9 and underscores
This commit is contained in:
N00MKRAD
2021-06-20 19:33:23 +02:00
parent c8551aabd3
commit bbeb6913ff
1 changed files with 23 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

23
Code/Main/AiModels.cs
View File

@@ -5,6 +5,7 @@ using System.Collections.Generic;
using System.IO;
using System.Linq;
using System.Text;
using System.Text.RegularExpressions;
using System.Threading.Tasks;
namespace Flowframes.Main
@@ -16,9 +17,31 @@ namespace Flowframes.Main
string pkgPath = Path.Combine(Paths.GetPkgPath(), ai.pkgDir);
string modelsFile = Path.Combine(pkgPath, "models.json");
ModelCollection modelCollection = new ModelCollection(ai, modelsFile);
foreach (string customModel in GetCustomModels(ai))
{
string name = customModel.Remove("_alpha").Remove("_custom");
bool alpha = customModel.Contains("_alpha");
modelCollection.models.Add(new ModelCollection.ModelInfo(ai, name, "Custom Model", customModel, alpha, false));
}
return modelCollection;
}
public static List<string> GetCustomModels(AI ai)
{
string pkgPath = Path.Combine(Paths.GetPkgPath(), ai.pkgDir);
List<string> custModels = new List<string>();
foreach (DirectoryInfo dir in new DirectoryInfo(pkgPath).GetDirectories())
{
if (dir.Name.EndsWith("_custom") && Regex.IsMatch(dir.Name, @"^[a-zA-Z0-9_]+$"))
custModels.Add(dir.Name);
}
return custModels;
}
public static ModelCollection.ModelInfo GetModelByName(AI ai, string modelName)
{
ModelCollection modelCollection = GetModels(ai);