mirror of
https://github.com/MindWorkAI/AI-Studio.git
synced 2026-10-17 16:58:00 +00:00
Move the model kinds into the rule engine
This commit is contained in:
16 files changed
+731
-10
No files matched your search
@@ -0,0 +1,36 @@
|
||||
using AIStudio.Provider;
|
||||
|
||||
namespace AIStudio.Models.Kinds;
|
||||
|
||||
/// <summary>
|
||||
/// The models which make video.
|
||||
/// </summary>
|
||||
/// <remarks>
|
||||
/// Two of these names have to stand as a name part of their own. "kling" taken as a plain substring
|
||||
/// also matches the organization Klingspor, the model Inkling, and the fine-tune
|
||||
/// Llama-2-7b-chat-klingon -- all of them models to chat with, which would vanish from the user's
|
||||
/// list. The models themselves are called kling-v1 and kling-video, where the name ends at a
|
||||
/// separator. Google's veo is the same story with an even shorter word.
|
||||
/// </remarks>
|
||||
public sealed class VideoGenerationModelsFamily : ModelFamily
|
||||
{
|
||||
/// <inheritdoc />
|
||||
public override ModelVendor Vendor => ModelVendor.UNKNOWN;
|
||||
|
||||
/// <inheritdoc />
|
||||
public override ModelSource Source => new("https://huggingface.co/models?pipeline_tag=text-to-video", new DateOnly(2026, 9, 12), "Ported from the video generation markers of Provider/ModelKindExtensions.cs, where veo carried a trailing hyphen to say the same thing a name part says here.");
|
||||
|
||||
/// <inheritdoc />
|
||||
protected override void Declare(ModelFamilyBuilder builder)
|
||||
{
|
||||
builder.Modifier("sora").AsSubstring().Kind(ModelKind.VIDEO_GENERATION);
|
||||
|
||||
builder.Modifier("runway").AsSubstring().Inherits();
|
||||
|
||||
builder.Modifier("hailuo").AsSubstring().Inherits();
|
||||
|
||||
builder.Modifier("veo").AsSegment().Inherits();
|
||||
|
||||
builder.Modifier("kling").AsSegment().Inherits();
|
||||
}
|
||||
}
|
||||
Reference in new issue
Block a user