rwkv.cpp(ggml) support

2023-12-12 20:29:55 +08:00
parent 6e29f97881
commit b14fbc29b7
26 changed files with 1234 additions and 102 deletions
--- a/frontend/src/utils/index.tsx
+++ b/frontend/src/utils/index.tsx
@@ -63,7 +63,7 @@ export async function refreshBuiltInModels(readCache: boolean = false) {
  return cache;
 }

-const modelSuffix = ['.pth', '.st', '.safetensors'];
+const modelSuffix = ['.pth', '.st', '.safetensors', '.bin'];

 export async function refreshLocalModels(cache: {
  models: ModelSourceItem[]