mirror of
https://github.com/ollama/ollama.git
synced 2026-02-05 05:03:21 -05:00
33 lines
627 B
Go
33 lines
627 B
Go
package runner
|
|
|
|
import (
|
|
"github.com/ollama/ollama/runner/llamarunner"
|
|
"github.com/ollama/ollama/runner/ollamarunner"
|
|
"github.com/ollama/ollama/x/mlxrunner"
|
|
)
|
|
|
|
func Execute(args []string) error {
|
|
if args[0] == "runner" {
|
|
args = args[1:]
|
|
}
|
|
|
|
var newRunner bool
|
|
var mlxRunner bool
|
|
if len(args) > 0 && args[0] == "--ollama-engine" {
|
|
args = args[1:]
|
|
newRunner = true
|
|
}
|
|
if len(args) > 0 && args[0] == "--mlx-engine" {
|
|
args = args[1:]
|
|
mlxRunner = true
|
|
}
|
|
|
|
if mlxRunner {
|
|
return mlxrunner.Execute(args)
|
|
} else if newRunner {
|
|
return ollamarunner.Execute(args)
|
|
} else {
|
|
return llamarunner.Execute(args)
|
|
}
|
|
}
|