modeldownloader

Documentation

Index

• type D
  • func New(modelDir string, s3Client s3Client) *D
  • func (d *D) Download(ctx context.Context, modelName string, resp *mv1.GetBaseModelPathResponse, ...) error

Types

type D

type D struct {
	// contains filtered or unexported fields
}

D is a downloader.

func New

func New(modelDir string, s3Client s3Client) *D

New returns a new Manager.

func (*D) Download

func (d *D) Download(
	ctx context.Context,
	modelName string,
	resp *mv1.GetBaseModelPathResponse,
	format mv1.ModelFormat,
	destPath string,
) error

Download downloads the model.