瀏覽代碼

Fix RWKV paths on Windows (attempt)

oobabooga 2 年之前
父節點
當前提交
449116a510
共有 1 個文件被更改,包括 2 次插入2 次删除
  1. 2 2
      modules/RWKV.py

+ 2 - 2
modules/RWKV.py

@@ -25,8 +25,8 @@ class RWKVModel:
     def from_pretrained(self, path, dtype="fp16", device="cuda"):
         tokenizer_path = Path(f"{path.parent}/20B_tokenizer.json")
 
-        model = RWKV(model=path.as_posix(), strategy=f'{device} {dtype}')
-        pipeline = PIPELINE(model, tokenizer_path.as_posix())
+        model = RWKV(model=os.path.abspath(path), strategy=f'{device} {dtype}')
+        pipeline = PIPELINE(model, os.path.abspath(tokenizer_path))
 
         result = self()
         result.pipeline = pipeline