add WebGPU Python Mode (https://github.com/cryscan/web-rwkv-py)

2023-12-14 18:37:07 +08:00
parent 01c95f5bc4
commit 0ddd2e9fea
16 changed files with 155 additions and 34 deletions
--- a/backend-python/main.py
+++ b/backend-python/main.py
@@ -37,6 +37,11 @@ def get_args(args: Union[Sequence[str], None] = None):
        action="store_true",
        help="whether to use rwkv.cpp (default: False)",
    )
+    group.add_argument(
+        "--webgpu",
+        action="store_true",
+        help="whether to use webgpu (default: False)",
+    )
    args = parser.parse_args(args)

    return args