models

Documentation

Index

• func DownloadFile(ctx context.Context, tenant, user, projectName, modelName, versionName string, ...) error
• func UploadFile(ctx context.Context, tenant, user, projectName, modelName, versionName string) error
• type ChunkInfo
• type Model

Functions

func DownloadFile

func DownloadFile(ctx context.Context, tenant, user, projectName, modelName, versionName string, model *Model) error

DownloadFile downloads the model file from harbor

func UploadFile

func UploadFile(ctx context.Context, tenant, user, projectName, modelName, versionName string) error

UploadFile uploads the model file to harbor

Types

type ChunkInfo

type ChunkInfo struct {
	Path      string
	FileName  string
	Content   io.ReadCloser
	TotalSize int64
	PartFrom  int64
	PartTo    int64
}

type Model

type Model struct {
	ModelName   string         `json:"modelName,omitempty"`
	ProjectName string         `json:"projectName,omitempty"`
	VersionName string         `json:"versionName,omitempty"`
	Description string         `json:"description,omitempty"`
	Format      string         `json:"format,omitempty"`
	FrameWork   string         `json:"framework,omitempty"`
	Inputs      []model.Tensor `json:"inputs,omitempty"`
	Outputs     []model.Tensor `json:"outputs,omitempty"`
}