oops

harrisonvanderbyl · Jan 9, 2024 · 70b03a8 · 70b03a8
1 parent 3f3c329
commit 70b03a8
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/godot-rwkv.h b/godot-rwkv.h
@@ -134,14 +134,14 @@ class GodotRWKV : public Resource {
 
 	}
 
-	void loadModel(String path, size_t NumThreads = 8) {
+	void loadModel(String path, int NumThreads = 8) {
 
 		if (NumThreads != 1 && NumThreads != 2 && NumThreads != 4 && NumThreads != 8) {
 			ERR_PRINT("NumThreads must be 8, 4, 2 or 1");
 			return;
 		}
 
-		model = new RWKV(std::string(path.utf8().get_data()), NumThreads);
+		model = new RWKV(std::string(path.utf8().get_data()), size_t(NumThreads))
 	};
 
 	void loadTokenizer(String path) {