Documentation
¶
Index ¶
- type FakeModelClient
- func (c *FakeModelClient) CreateBaseModel(ctx context.Context, in *mv1.CreateBaseModelRequest, opts ...grpc.CallOption) (*mv1.BaseModel, error)
- func (c *FakeModelClient) GetBaseModelPath(ctx context.Context, in *mv1.GetBaseModelPathRequest, opts ...grpc.CallOption) (*mv1.GetBaseModelPathResponse, error)
- type HuggingFaceDownloader
- type L
- type ModelClient
- type ModelDownloader
- type NoopModelDownloader
- type NoopS3Client
- type S3Client
- type S3Downloader
Constants ¶
This section is empty.
Variables ¶
This section is empty.
Functions ¶
This section is empty.
Types ¶
type FakeModelClient ¶
type FakeModelClient struct {
// contains filtered or unexported fields
}
FakeModelClient is a fake model client.
func NewFakeModelClient ¶
func NewFakeModelClient() *FakeModelClient
NewFakeModelClient creates a fake model client.
func (*FakeModelClient) CreateBaseModel ¶
func (c *FakeModelClient) CreateBaseModel(ctx context.Context, in *mv1.CreateBaseModelRequest, opts ...grpc.CallOption) (*mv1.BaseModel, error)
CreateBaseModel creates a base model.
func (*FakeModelClient) GetBaseModelPath ¶
func (c *FakeModelClient) GetBaseModelPath(ctx context.Context, in *mv1.GetBaseModelPathRequest, opts ...grpc.CallOption) (*mv1.GetBaseModelPathResponse, error)
GetBaseModelPath gets the path of a base model.
type HuggingFaceDownloader ¶
type HuggingFaceDownloader struct {
// contains filtered or unexported fields
}
HuggingFaceDownloader downloads models from Hugging Face.
func NewHuggingFaceDownloader ¶
func NewHuggingFaceDownloader(cacheDir string) *HuggingFaceDownloader
NewHuggingFaceDownloader creates a new HuggingFaceDownloader.
type L ¶
type L struct {
// contains filtered or unexported fields
}
L is a loader.
func New ¶
func New( baseModels []string, objectStorePathPrefix string, modelDownloader ModelDownloader, s3Client S3Client, modelClient ModelClient, ) *L
New creates a new loader.
func (*L) LoadBaseModels ¶
LoadBaseModels loads the base models.
type ModelClient ¶
type ModelClient interface {
CreateBaseModel(ctx context.Context, in *mv1.CreateBaseModelRequest, opts ...grpc.CallOption) (*mv1.BaseModel, error)
GetBaseModelPath(ctx context.Context, in *mv1.GetBaseModelPathRequest, opts ...grpc.CallOption) (*mv1.GetBaseModelPathResponse, error)
}
ModelClient is an interface for the model client.
type ModelDownloader ¶
type ModelDownloader interface {
// contains filtered or unexported methods
}
ModelDownloader is an interface for downloading a model.
type NoopModelDownloader ¶
type NoopModelDownloader struct {
}
NoopModelDownloader is a no-op model downloader.
type S3Downloader ¶
type S3Downloader struct {
// contains filtered or unexported fields
}
S3Downloader downloads models from S3.
func NewS3Downloader ¶
func NewS3Downloader(s3Client s3Client, pathPrefix string) *S3Downloader
NewS3Downloader returns a new S3Downloader.
Source Files
Click to show internal directories.
Click to hide internal directories.