File tree Expand file tree Collapse file tree
Expand file tree Collapse file tree Original file line number Diff line number Diff 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 ,
You can’t perform that action at this time.
0 commit comments