From a0e06bd306e8ff2a4e25d0ee6c80012ddb4d3453 Mon Sep 17 00:00:00 2001 From: N00MKRAD Date: Mon, 5 Apr 2021 20:06:19 +0200 Subject: [PATCH] Fix GPU ID defaults --- Code/IO/Config.cs | 2 ++ 1 file changed, 2 insertions(+) diff --git a/Code/IO/Config.cs b/Code/IO/Config.cs index ae36c0f..798c6d4 100644 --- a/Code/IO/Config.cs +++ b/Code/IO/Config.cs @@ -164,6 +164,8 @@ namespace Flowframes.IO // AI if (key == "uhdThresh") return WriteDefault(key, "1600"); if (key == "rifeCudaFp16") return WriteDefault(key, NvApi.HasTensorCores().ToString()); + if (key == "torchGpus") return WriteDefault(key, "0"); + if (key == "ncnnGpus") return WriteDefault(key, "0"); if (key == "ncnnThreads") return WriteDefault(key, "1"); if (key == "dainNcnnTilesize") return WriteDefault(key, "768"); // Debug / Other / Experimental