ollama

Documentation

Index

• func CreateModelfile(filePath string, modelName string, spec *ModelSpec, ...) error
• func ModelName(modelID string) string
• func ModelfilePath(modelDir string, modelName string) string
• func WriteModelfile(modelName string, spec *ModelSpec, contextLengthsByModelID map[string]int, ...) error
• type ModelSpec

Functions

func CreateModelfile

func CreateModelfile(
	filePath string,
	modelName string,
	spec *ModelSpec,
	contextLengthsByModelID map[string]int,
) error

CreateModelfile creates a new model file.

func ModelName

func ModelName(modelID string) string

ModelName returns the Ollama model name from the model ID used in LLM Operator.

Ollama does not accept more than two ":" while the model ID of the fine-tuning jobs is be "ft:<base-model>:<suffix>".

func ModelfilePath

func ModelfilePath(modelDir string, modelName string) string

ModelfilePath returns the model file path.

func WriteModelfile

func WriteModelfile(
	modelName string,
	spec *ModelSpec,
	contextLengthsByModelID map[string]int,
	file *os.File,
) error

WriteModelfile writes the model file.

Types

type ModelSpec

type ModelSpec struct {
	From        string
	AdapterPath string
}

ModelSpec is the specification for a new model.