ソースを参照

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