AXKuhta / rwkv-onnx-dml

Run ONNX RWKV-v4 models with GPU acceleration using DirectML [Windows], or just on CPU [Windows AND Linux]; Limited to 430M model at this time because of .onnx 2GB file size limitation

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

AXKuhta/rwkv-onnx-dml Watchers