modeldownloader

Documentation

Index

• func AdapterFilePath(modelDir, modelID string) (string, error)
• func ModelFilePath(modelDir, modelID string, format mv1.ModelFormat) (string, error)
• type D
  • func New(modelDir string, s3Client s3Client) *D
  • func (d *D) Download(ctx context.Context, modelID string, srcPath string, format mv1.ModelFormat, ...) error
  • func (d *D) DownloadAdapterOfGGUF(ctx context.Context, modelID string, resp *mv1.GetModelPathResponse) error

Functions

func AdapterFilePath

func AdapterFilePath(modelDir, modelID string) (string, error)

AdapterFilePath returns the file path of the adapter.

func ModelFilePath

func ModelFilePath(modelDir, modelID string, format mv1.ModelFormat) (string, error)

ModelFilePath returns the file path of the model.

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,
	modelID string,
	srcPath string,
	format mv1.ModelFormat,
	adapterType mv1.AdapterType,
) error

Download downloads the model.

func (*D) DownloadAdapterOfGGUF

func (d *D) DownloadAdapterOfGGUF(
	ctx context.Context,
	modelID string,
	resp *mv1.GetModelPathResponse,
) error

DownloadAdapterOfGGUF downloads the adapter.