Hazzzardous commited on
Commit
b24db59
1 Parent(s): ac182ec

Update config.py

Browse files
Files changed (1) hide show
  1. config.py +4 -4
config.py CHANGED
@@ -19,14 +19,14 @@ quantized = {
19
 
20
 
21
  config = {
22
- "path":"https://huggingface.co/BlinkDL/rwkv-4-pile-7b/resolve/main/RWKV-4-Pile-7B-Instruct-test1-20230124.pth",
23
  "mode":TORCH,
24
- "runtimedtype":torch.bfloat16,
25
  "useGPU":torch.cuda.is_available(),
26
- "dtype":torch.bfloat16
27
  }
28
 
29
- title = "RWKV-4 (7B Instruct)"
30
 
31
  # RWKV 1B5 instruct model quantized
32
  # Approximate
 
19
 
20
 
21
  config = {
22
+ "path":"https://huggingface.co/BlinkDL/rwkv-4-pile-1b5/resolve/main/RWKV-4-Pile-1B5-Instruct-test2-20230209.pth",
23
  "mode":TORCH,
24
+ "runtimedtype":torch.float32,
25
  "useGPU":torch.cuda.is_available(),
26
+ "dtype":torch.float32
27
  }
28
 
29
+ title = "RWKV-4 (1B5 Instruct 2)"
30
 
31
  # RWKV 1B5 instruct model quantized
32
  # Approximate