瀏覽代碼

no gpu if vram < 2GB

Michael Yang 1 年之前
父節點
當前提交
811c3d1900
共有 1 個文件被更改,包括 5 次插入0 次删除
  1. 5 0
      llm/llama.go

+ 5 - 0
llm/llama.go

@@ -215,6 +215,11 @@ func CheckVRAM() (int64, error) {
 		free += vram
 		free += vram
 	}
 	}
 
 
+	if free*1024*1024 < 2*1000*1000*1000 {
+		log.Printf("less than 2 GB VRAM available, falling back to CPU only")
+		free = 0
+	}
+
 	return free, nil
 	return free, nil
 }
 }