services

Documentation

Index

• type ModelService
  • func NewModelService(r repository.ModelRepository, t triton.TritonService) ModelService

Types

type ModelService

type ModelService interface {
	CreateModel(model *models.Model) (*models.Model, error)
	GetModelByName(namespace string, modelName string) (models.Model, error)
	CreateVersion(version models.Version) (models.Version, error)
	GetModelVersion(modelId uint64, version uint64) (models.Version, error)
	GetModelVersions(modelId uint64) ([]models.Version, error)
	GetModelVersionLatest(modelId uint64) (models.Version, error)
	GetFullModelData(namespace string, modelName string) (*model.Model, error)
	ModelInfer(namespace string, modelName string, version uint64, imgsBytes [][]byte, task model.Model_Task) (interface{}, error)
	CreateModelBinaryFileUpload(namespace string, createdModel *models.Model) (*model.Model, error)
	ListModels(namespace string) ([]*model.Model, error)
	UpdateModelVersion(namespace string, updatedInfo *model.UpdateModelVersionRequest) (*model.ModelVersion, error)
	DeleteModel(namespace string, modelName string) error
	DeleteModelVersion(namespace string, modelName string, version uint64) error
}

func NewModelService

func NewModelService(r repository.ModelRepository, t triton.TritonService) ModelService