Documentation
¶
Index ¶
Constants ¶
This section is empty.
Variables ¶
This section is empty.
Functions ¶
This section is empty.
Types ¶
type AIModel ¶
type AIModel struct {
ID string `yaml:"-"`
Name string `yaml:"name"`
ModuleDescription string `yaml:"description"`
Runner string `yaml:"runner"`
Bricks []string `yaml:"bricks,omitempty"`
ModelLabels []string `yaml:"model_labels,omitempty"`
Metadata map[string]string `yaml:"metadata,omitempty"`
ModelConfiguration map[string]string `yaml:"model_configuration,omitempty"`
}
type ModelsIndex ¶
type ModelsIndex struct {
// contains filtered or unexported fields
}
func GenerateModelsIndexFromFile ¶
func GenerateModelsIndexFromFile(dir *paths.Path) (*ModelsIndex, error)
func (*ModelsIndex) GetModelByID ¶
func (m *ModelsIndex) GetModelByID(id string) (*AIModel, bool)
func (*ModelsIndex) GetModels ¶
func (m *ModelsIndex) GetModels() []AIModel
func (*ModelsIndex) GetModelsByBrick ¶
func (m *ModelsIndex) GetModelsByBrick(brick string) []AIModel
func (*ModelsIndex) GetModelsByBricks ¶
func (m *ModelsIndex) GetModelsByBricks(bricks []string) []AIModel
Click to show internal directories.
Click to hide internal directories.