浏览代码

Merge pull request #4743 from 5E-324/add-num_gpu

fix: Changes to num_gpu made in Settings > General won't be saved
Timothy Jaeryang Baek 8 月之前
父节点
当前提交
27109d22e4
共有 1 个文件被更改,包括 4 次插入2 次删除
  1. 4 2
      src/lib/components/chat/Settings/General.svelte

+ 4 - 2
src/lib/components/chat/Settings/General.svelte

@@ -59,7 +59,8 @@
 		num_ctx: null,
 		num_batch: null,
 		num_keep: null,
-		max_tokens: null
+		max_tokens: null,
+		num_gpu: null
 	};
 
 	const toggleRequestFormat = async () => {
@@ -321,7 +322,8 @@
 						max_tokens: params.max_tokens !== null ? params.max_tokens : undefined,
 						use_mmap: params.use_mmap !== null ? params.use_mmap : undefined,
 						use_mlock: params.use_mlock !== null ? params.use_mlock : undefined,
-						num_thread: params.num_thread !== null ? params.num_thread : undefined
+						num_thread: params.num_thread !== null ? params.num_thread : undefined,
+						num_gpu: params.num_gpu !== null ? params.num_gpu : undefined
 					},
 					keepAlive: keepAlive ? (isNaN(keepAlive) ? keepAlive : parseInt(keepAlive)) : undefined
 				});