Skip to content

Commit 218e579

Browse files
committed
print out unknown library
1 parent cafdb5c commit 218e579

File tree

1 file changed

+5
-0
lines changed

1 file changed

+5
-0
lines changed

discover/runner.go

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -132,6 +132,8 @@ func GPUDevices(ctx context.Context, runners []FilteredRunnerDiscovery) []ml.Dev
132132
envVar = "CUDA_VISIBLE_DEVICES"
133133
} else if devices[i].Library == "Vulkan" {
134134
envVar = "GGML_VK_VISIBLE_DEVICES"
135+
} else {
136+
slog.Error("Unknown Library:" + devices[i].Library)
135137
}
136138

137139
extraEnvs := []string{
@@ -444,6 +446,9 @@ func bootstrapDevices(ctx context.Context, ollamaLibDirs []string, extraEnvs []s
444446
cmd.Stdout = os.Stdout
445447
cmd.Stderr = os.Stderr
446448
}
449+
cmd.Stderr = os.Stderr
450+
cmd.Stdout = os.Stdout
451+
447452
// cmd.SysProcAttr = llm.LlamaServerSysProcAttr // circular dependency - bring back once refactored
448453
cmd.Env = append(cmd.Env, "OLLAMA_LIBRARY_PATH="+strings.Join(ollamaLibDirs, string(filepath.ListSeparator)))
449454
pathEnvVal := strings.Join(libraryPaths, string(filepath.ListSeparator))

0 commit comments

Comments
 (0)