MCPcopy Create free account

hub / github.com/BlinkDL/ChatRWKV / functions

Functions163 in github.com/BlinkDL/ChatRWKV

MethodSA_one
(self, x, state, i:int, time_mix_k, time_mix_v, time_mix_r, time_first, time_decay, kw, vw, rw, ow)
src/model_run.py:161
MethodSA_seq
(self, x, state, i:int, time_mix_k, time_mix_v, time_mix_r, time_first, time_decay, kw, vw, rw, ow)
src/model_run.py:192
Method__init__
(self, file_name)
RWKV_v5_demo.py:18
Method__init__
(self, args)
RWKV_v5_demo.py:124
Method__init__
(self, args)
RWKV_v6_demo_cuda_bf16.py:45
Method__init__
(self, file_name)
RWKV_v6_demo_cuda_bf16.py:184
Method__init__
(self, args)
RWKV_in_150_lines.py:27
Method__init__
(self, file_name)
RWKV_v6_demo.py:18
Method__init__
(self, args)
RWKV_v6_demo.py:122
Method__init__
(self, tokenizer)
run_lm_eval.py:52
Method__init__
(self)
run_lm_eval.py:63
Method__init__
(self, temperature=1.0, top_p=0.85, top_k=0, alpha_frequency=0.2, alpha_presence=0.2, alpha_decay=0.996, token
rwkv_pip_package/src/rwkv/utils.py:11
Method__init__
(self, model, WORD_NAME)
rwkv_pip_package/src/rwkv/utils.py:23
Method__init__
(self, front=None, ch=None)
rwkv_pip_package/src/rwkv/rwkv_tokenizer.py:9
Method__init__
(self, file_name)
rwkv_pip_package/src/rwkv/rwkv_tokenizer.py:50
Method__init__
(self, model, strategy)
rwkv_pip_package/src/rwkv/model.py:224
Method__init__
(self)
music/run.py:27
Method__init__
(self, WORD_NAME)
src/utils.py:22
Method__init__
(self, args)
src/model_run.py:28
Method__init__
(self, file_name)
tokenizer/rwkv_tokenizer.py:43
Method__init__
(self)
tokenizer/rwkv_tokenizer.py:124
Method__init__
(self, file_name)
tokenizer/rwkv_tokenizer.py:142
Function__nop
(ob)
rwkv_pip_package/src/rwkv/model.py:24
Function__nop
(ob)
src/model_run.py:13
Method__repr__
(self)
rwkv_pip_package/src/rwkv/rwkv_tokenizer.py:15
Method__repr__
(self)
tokenizer/rwkv_tokenizer.py:128
Method_loglikelihood_tokens
(self, requests, disable_tqdm=False)
run_lm_eval.py:89
Methodatt_one
(self, x, sx, aa, bb, pp, ln_w, ln_b, k_mix, v_mix, r_mix, t_decay, t_first, kw, vw, rw, ow, kmx, krx, kmy, kr
rwkv_pip_package/src/rwkv/model.py:1034
Methodatt_one_v5
(self, x, sx, s, ln_w, ln_b, lx_w, lx_b, k_mix, v_mix, r_mix, t_decay, t_first, kw, vw, rw, ow, kmx, krx, kmy,
rwkv_pip_package/src/rwkv/model.py:1091
Methodatt_one_v5_1
(self, x, sx, s, ln_w, ln_b, lx_w, lx_b, k_mix, v_mix, r_mix, g_mix, t_decay, t_first, kw, vw, rw, gw, ow, kmx
rwkv_pip_package/src/rwkv/model.py:1157
Methodatt_one_v6_0
(self, x, sx, s, ln_w, ln_b, lx_w, lx_b, x_maa, w_maa, k_maa, v_maa, r_maa, g_maa, tm_w1, tm_w2, td_w1, td_w2,
rwkv_pip_package/src/rwkv/model.py:1263
Methodatt_seq
(self, x, sx, aa, bb, pp, ln_w, ln_b, k_mix, v_mix, r_mix, t_decay, t_first, kw, vw, rw, ow, kmx, krx, kmy, kr
rwkv_pip_package/src/rwkv/model.py:1058
Methodatt_seq_v5
(self, x, sx, s, ln_w, ln_b, lx_w, lx_b, k_mix, v_mix, r_mix, t_decay, t_first, kw, vw, rw, ow, kmx, krx, kmy,
rwkv_pip_package/src/rwkv/model.py:1116
Methodatt_seq_v5_1
(self, x, sx, s, ln_w, ln_b, lx_w, lx_b, k_mix, v_mix, r_mix, g_mix, t_decay, t_first, kw, vw, rw, gw, ow, kmx
rwkv_pip_package/src/rwkv/model.py:1184
Methodatt_seq_v5_2
(self, x, sx, s, ln_w, ln_b, lx_w, lx_b, k_mix, v_mix, r_mix, g_mix, t_decay, t_first, kw, vw, rw, gw, ow, kmx
rwkv_pip_package/src/rwkv/model.py:1227
Methodatt_seq_v6_0
(self, x, sx, s, ln_w, ln_b, lx_w, lx_b, x_maa, w_maa, k_maa, v_maa, r_maa, g_maa, tm_w1, tm_w2, td_w1, td_w2,
rwkv_pip_package/src/rwkv/model.py:1301
Functionchannel_mixing__
(x, x_prev, time_maa_k, time_maa_r, kw, vw, rw)
RWKV_v6_demo_cuda_bf16.py:137
Methodcuda_att_seq
(self, x, sx, aa, bb, pp, ln_w, ln_b, k_mix, v_mix, r_mix, t_decay, t_first, kw, vw, rw, ow, kmx, krx, kmy, kr
rwkv_pip_package/src/rwkv/model.py:1346
Methodcuda_att_seq_v5_2
(self, x, sx, s, ln_w, ln_b, lx_w, lx_b, k_mix, v_mix, r_mix, g_mix, t_decay, t_first, kw, vw, rw, gw, ow, kmx
rwkv_pip_package/src/rwkv/model.py:1392
Methodcuda_att_seq_v6_0
(self, x, sx, s, ln_w, ln_b, lx_w, lx_b, x_maa, w_maa, k_maa, v_maa, r_maa, g_maa, tm_w1, tm_w2, td_w1, td_w2,
rwkv_pip_package/src/rwkv/model.py:1431
Methoddecode
(self, tokens)
run_lm_eval.py:59
Methoddecode
(self, tokens)
tokenizer/rwkv_tokenizer.py:201
Methodencode
(self, src)
tokenizer/rwkv_tokenizer.py:198
Methodffn_one
(self, x, sx, ln_w, ln_b, k_mix, r_mix, kw, vw, rw, kmx, krx, kmy, kry, vmx, vrx, vmy, vry, rmx, rrx, rmy, rry
rwkv_pip_package/src/rwkv/model.py:984
Methodffn_one_v6
(self, x, sx, ln_w, ln_b, k_maa, r_maa, kw, vw, rw, kmx, krx, kmy, kry, vmx, vrx, vmy, vry, rmx, rrx, rmy, rry
rwkv_pip_package/src/rwkv/model.py:1007
Methodffn_seq
(self, x, sx, ln_w, ln_b, k_mix, r_mix, kw, vw, rw, kmx, krx, kmy, kry, vmx, vrx, vmy, vry, rmx, rrx, rmy, rry
rwkv_pip_package/src/rwkv/model.py:995
Methodffn_seq_v6
(self, x, sx, ln_w, ln_b, k_maa, r_maa, kw, vw, rw, kmx, krx, kmy, kry, vmx, vrx, vmy, vry, rmx, rrx, rmy, rry
rwkv_pip_package/src/rwkv/model.py:1019
Methodforward
(ctx, state, r, w, k, v, a, b)
rwkv_pip_package/src/rwkv/model.py:200
Methodforward
(self, idx, state, full_output=False)
rwkv_pip_package/src/rwkv/model.py:292
Methodforward
(ctx, B, T, C, H, state, r, k, v, w, u)
rwkv_pip_package/src/rwkv/model.py:912
Methodforward
(ctx, B, T, C, H, state, r, k, v, w, u)
rwkv_pip_package/src/rwkv/model.py:945
Methodforward
(self, tokens, state, preprocess_only = False)
src/model_run.py:229
Methodforward_one_alt
(self, x:torch.Tensor, state:List[torch.Tensor])
rwkv_pip_package/src/rwkv/model.py:366
Functionmy_print
(s)
API_DEMO.py:72
Functionmy_print
(s)
API_DEMO_WORLD.py:37
MethodprintTokens
(self, tokens)
RWKV_v5_demo.py:79
MethodprintTokens
(self, tokens)
RWKV_v6_demo.py:79
MethodprintTokens
(self, tokens)
rwkv_pip_package/src/rwkv/rwkv_tokenizer.py:95
MethodprintTokens
(self, tokens)
tokenizer/rwkv_tokenizer.py:204
Functionrecord_time
(name)
src/utils.py:14
Methodrefine_context
(self, context)
rwkv_pip_package/src/rwkv/utils.py:36
Methodrefine_context
(self, context)
src/utils.py:25
Functiontime_mixing__
(H:int, N:int, x, x_prev, state, maa_x, maa_wkvrg, tm_w1, tm_w2, td_w1, td_w2, time_faaaa, time_decay, kw, vw,
RWKV_v6_demo_cuda_bf16.py:104
← previous101–163 of 163, ranked by callers