diff --git a/modules/RWKV.py b/modules/RWKV.py index 88f1ec2..ee8d76f 100644 --- a/modules/RWKV.py +++ b/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