Skip to content

Commit e10c8be

Browse files
[pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
1 parent aa8592a commit e10c8be

1 file changed

Lines changed: 2 additions & 2 deletions

File tree

elk/rwkv_lm/rwkv_hf.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -25,8 +25,8 @@ def __init__(self):
2525
super().__init__(RWKVConfig())
2626
# weights_path = "/home/kyle/HF-MODEL/rwkv-4-pile-1b5/models--BlinkDL--rwkv-4-pile-1b5/snapshots/6ea995eaa87a17af560c9b41ce1a3d92355c5a49/RWKV-4-Pile-1B5-20220903-8040.pth"
2727
# weights_path = "/home/kyle/HF-MODEL/rwkv-4-pile-14b/models--BlinkDL--rwkv-4-pile-14b/snapshots/939b6851f96122b7b49bd00d446b3b49481214dd/RWKV-4-Pile-14B-20230213-8019.pth"
28-
weights_path= "/home/kyobrien/.cache/huggingface/hub/models--BlinkDL--rwkv-4-pile-1b5/snapshots/6ea995eaa87a17af560c9b41ce1a3d92355c5a49/RWKV-4-Pile-1B5-20220903-8040.pth"
29-
self.model = RWKV(model=weights_path, strategy='cuda bf16')
28+
weights_path = "/home/kyobrien/.cache/huggingface/hub/models--BlinkDL--rwkv-4-pile-1b5/snapshots/6ea995eaa87a17af560c9b41ce1a3d92355c5a49/RWKV-4-Pile-1B5-20220903-8040.pth"
29+
self.model = RWKV(model=weights_path, strategy="cuda bf16")
3030

3131
def forward(
3232
self,

0 commit comments

Comments
 (0)