Update gpu.go

changed AssetsDir() to PayloadsDir() to match upstream
ollama · Mar 24, 2024 · e27f021 · e27f021
1 parent b8731c3
commit e27f021
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/gpu/gpu.go b/gpu/gpu.go
@@ -88,7 +88,7 @@ func initGPUHandles() {
 	var cudartMgmtName string
 	var cudartMgmtPatterns []string
 
-	tmpDir, _ := AssetsDir()
+	tmpDir, _ := PayloadsDir()
 	switch runtime.GOOS {
 	case "windows":
 		nvmlMgmtName = "nvml.dll"