github-starred/ollama
2
0
Fork 0
mirror of https://github.com/ollama/ollama.git synced 2026-03-11 17:34:04 -05:00
Code Issues Packages Projects Releases Wiki Activity

mlx: prevent remote creation mismatch (#14651)

Browse Source 
If the user is pointing at a remote OLLAMA_HOST, fail experimental safetensor
based create operations as we only support local creation currently.
This commit is contained in:
Daniel Hiltgen
2026-03-05 14:59:00 -08:00
committed by GitHub
parent 9896e3627f
commit 6928630601
1 changed files with 12 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
cmd/cmd.go
View File

@@ -145,6 +145,12 @@ func CreateHandler(cmd *cobra.Command, args []string) error {
// Check for --experimental flag for safetensors model creation
experimental, _ := cmd.Flags().GetBool("experimental")
if experimental {
host := envconfig.Host()
h, _, _ := net.SplitHostPort(host.Host)
ip := net.ParseIP(h)
if ip == nil || (!ip.IsLoopback() && !ip.IsUnspecified()) {
return errors.New("remote safetensor model creation not yet supported")
}
// Get Modelfile content - either from -f flag or default to "FROM ."
var reader io.Reader
filename, err := getModelfileName(cmd)
@@ -214,6 +220,12 @@ func CreateHandler(cmd *cobra.Command, args []string) error {
if filename == "" {
// No Modelfile found - check if current directory is an image gen model
if create.IsTensorModelDir(".") {
host := envconfig.Host()
h, _, _ := net.SplitHostPort(host.Host)
ip := net.ParseIP(h)
if ip == nil || (!ip.IsLoopback() && !ip.IsUnspecified()) {
return errors.New("remote safetensor model creation not yet supported")
}
quantize, _ := cmd.Flags().GetString("quantize")
return xcreateclient.CreateModel(xcreateclient.CreateOptions{
ModelName: modelName,