[pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci
EleutherAI · Apr 23, 2023 · 76b0c85 · 76b0c85
1 parent aa8592a
commit 76b0c85
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/elk/rwkv_lm/rwkv_hf.py b/elk/rwkv_lm/rwkv_hf.py
@@ -25,8 +25,8 @@ def __init__(self):
  super().__init__(RWKVConfig())
  # 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"
  # 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"
- weights_path= "/home/kyobrien/.cache/huggingface/hub/models--BlinkDL--rwkv-4-pile-1b5/snapshots/6ea995eaa87a17af560c9b41ce1a3d92355c5a49/RWKV-4-Pile-1B5-20220903-8040.pth"
- self.model = RWKV(model=weights_path, strategy='cuda bf16')
+ weights_path = "/home/kyobrien/.cache/huggingface/hub/models--BlinkDL--rwkv-4-pile-1b5/snapshots/6ea995eaa87a17af560c9b41ce1a3d92355c5a49/RWKV-4-Pile-1B5-20220903-8040.pth"
+ self.model = RWKV(model=weights_path, strategy="cuda bf16")
 
  def forward(
  self,