MCPcopy Create free account

hub / github.com/BlinkDL/ChatRWKV / functions

Functions163 in github.com/BlinkDL/ChatRWKV

↓ 71 callersFunctionmatmul
(a, b, mx: Optional[torch.Tensor]=None, rx: Optional[torch.Tensor]=None, my: Optional[torch.Tensor]=None, ry:
rwkv_pip_package/src/rwkv/model.py:127
↓ 48 callersMethodlayer_norm
(self, x, w)
RWKV_v5_demo.py:155
↓ 35 callersMethodforward
(self, token, state)
RWKV_v5_demo.py:198
↓ 24 callersMethodencode
(self, x)
src/utils.py:35
↓ 14 callersMethoddecode
(self, x)
src/utils.py:38
↓ 12 callersFunctionsave_all_stat
(srv, name, last_out)
v2/chat.py:167
↓ 10 callersFunctionrun_rnn
(tokens, newline_adj = 0)
v2/chat.py:149
↓ 10 callersMethodsample_logits
(self, logits, x, ctx_len, temperature=1.0, top_p=1.0)
src/utils.py:41
↓ 10 callersFunctionsave_all_stat
(srv, name, last_out)
chat.py:249
↓ 8 callersMethodencode
(self, src: str)
tokenizer/rwkv_tokenizer.py:98
↓ 8 callersFunctionrun_rnn
(tokens, newline_adj = 0)
chat.py:231
↓ 7 callersMethoddecode
(self, tokens)
tokenizer/rwkv_tokenizer.py:101
↓ 6 callersFunctionload_all_stat
(srv, name)
chat.py:256
↓ 6 callersFunctionload_all_stat
(srv, name)
v2/chat.py:174
↓ 5 callersMethodLN
(self, x, w)
src/model_run.py:131
↓ 5 callersMethodadd
(self, key:bytes, val:int)
tokenizer/rwkv_tokenizer.py:131
↓ 4 callersFunctionRWKV_x070_CMix_one
(x, x_prev, x_k, K_, V_, semb_, s1_, s2_, s0_)
rwkv_pip_package/src/rwkv/model.py:578
↓ 4 callersFunctionRWKV_x070_TMix_one
(layer_id: int, H:int, N:int, x, x_prev, v_first, state, x_r, x_w, x_k, x_v, x_a, x_g, w0, w1, w2, a0, a1, a2,
rwkv_pip_package/src/rwkv/model.py:492
↓ 4 callersMethodlayer_norm
(self, x, w)
RWKV_in_150_lines.py:54
↓ 4 callersMethodlayer_norm
(self, x, w)
RWKV_v6_demo.py:153
↓ 3 callersFunctioncfg_generator
(with_ctx, without_ctx, cfg_factor=1)
API_DEMO_WORLD.py:137
↓ 2 callersFunctionRWKV_x070_CMix_seq
(x, x_prev, x_k, K_, V_, semb_, s1_, s2_, s0_)
rwkv_pip_package/src/rwkv/model.py:587
↓ 2 callersFunctionRWKV_x070_TMix_seq
(layer_id: int, H:int, N:int, x, x_prev, v_first, state, x_r, x_w, x_k, x_v, x_a, x_g, w0, w1, w2, a0, a1, a2,
rwkv_pip_package/src/rwkv/model.py:520
↓ 2 callersFunctionbenchmark
(XXX)
tokenizer/rwkv_tokenizer.py:247
↓ 2 callersMethoddecode
(self, tokens)
RWKV_v5_demo.py:76
↓ 2 callersMethoddecode
(self, tokens)
RWKV_v6_demo.py:76
↓ 2 callersMethoddecode
(self, x)
rwkv_pip_package/src/rwkv/utils.py:52
↓ 2 callersMethodencode
(self, src: str)
RWKV_v5_demo.py:73
↓ 2 callersMethodencode
(self, src: str)
RWKV_v6_demo_cuda_bf16.py:239
↓ 2 callersMethodencode
(self, src: str)
RWKV_v6_demo.py:73
↓ 2 callersFunctionfix_tokens
(tokens)
v2/chat.py:182
↓ 2 callersMethodforward
(self, token:int, state:List[torch.Tensor])
RWKV_v6_demo_cuda_bf16.py:75
↓ 2 callersMethodforward
(self, token, state)
RWKV_in_150_lines.py:95
↓ 2 callersMethodforward
(self, token, state)
RWKV_v6_demo.py:207
↓ 2 callersMethodforward_one
(self, idx:int, state:List[torch.Tensor])
rwkv_pip_package/src/rwkv/model.py:306
↓ 2 callersMethodgenerate
(self, ctx, token_count=100, args=PIPELINE_ARGS(), callback=None, state=None)
rwkv_pip_package/src/rwkv/utils.py:90
↓ 2 callersFunctionload_prompt
(PROMPT_FILE)
v2/chat.py:110
↓ 2 callersFunctionmm8_one
(x, w, mx, rx, my, ry)
rwkv_pip_package/src/rwkv/model.py:108
↓ 2 callersFunctionmm8_seq
(x, w, mx, rx, my, ry)
rwkv_pip_package/src/rwkv/model.py:101
↓ 2 callersFunctionrecord_time
(name)
v2/benchmark.py:92
↓ 2 callersFunctionrun_rnn
(ctx)
API_DEMO_CHAT.py:75
↓ 2 callersFunctiontorch_mm8_one
(x, w, mx, rx, my, ry)
rwkv_pip_package/src/rwkv/model.py:96
↓ 2 callersFunctiontorch_mm8_seq
(x, w, mx, rx, my, ry)
rwkv_pip_package/src/rwkv/model.py:92
↓ 2 callersMethodv5_2_after
(self, t_decay, out, s, x, xxx, g, lx_w, lx_b, ow, omx, orx, omy, ory)
rwkv_pip_package/src/rwkv/model.py:1379
↓ 1 callersMethodRUN_RWKV_5
(self, B, T, C, H, state, r, k, v, w, u)
rwkv_pip_package/src/rwkv/model.py:975
↓ 1 callersMethodRUN_RWKV_6
(self, B, T, C, H, state, r, k, v, w, u)
rwkv_pip_package/src/rwkv/model.py:978
↓ 1 callersFunctionRWKV7_OP
(state, r, w, k, v, a, b)
rwkv_pip_package/src/rwkv/model.py:218
↓ 1 callersMethod__init__
(self, model, strategy, verbose = True, convert_and_save_and_exit = None)
rwkv_pip_package/src/rwkv/model.py:613
↓ 1 callersMethodadd
(self, key:bytes, idx:int=0, val=None)
rwkv_pip_package/src/rwkv/rwkv_tokenizer.py:24
↓ 1 callersMethodchannel_mixing
(self, x, state, i:int, time_mix_k, time_mix_r, kw, vw, rw)
RWKV_v5_demo.py:159
↓ 1 callersMethodchannel_mixing
(self, x, state, i:int, time_mix_k, time_mix_r, kw, vw, rw)
RWKV_in_150_lines.py:58
↓ 1 callersMethodchannel_mixing
(self, x, state, i:int, time_maa_k, time_maa_r, kw, vw, rw)
RWKV_v6_demo.py:157
↓ 1 callersFunctioncuda_mm8_one
(N: int, M: int, x, w, mx, rx, my, ry)
rwkv_pip_package/src/rwkv/model.py:76
↓ 1 callersFunctioncuda_mm8_seq
(B: int, N: int, M: int, x, w, mx, rx, my, ry)
rwkv_pip_package/src/rwkv/model.py:64
↓ 1 callersFunctioncuda_wkv
(T: int, C: int, w, u, k, v, aa, bb, pp)
rwkv_pip_package/src/rwkv/model.py:52
↓ 1 callersMethoddecode
(self, tokens)
RWKV_v6_demo_cuda_bf16.py:242
↓ 1 callersMethoddecode
(self, tokens)
rwkv_pip_package/src/rwkv/rwkv_tokenizer.py:89
↓ 1 callersMethoddecode
(self, ids)
music/run.py:34
↓ 1 callersMethoddecodeBytes
(self, tokens)
RWKV_v5_demo.py:70
↓ 1 callersMethoddecodeBytes
(self, tokens)
RWKV_v6_demo_cuda_bf16.py:236
↓ 1 callersMethoddecodeBytes
(self, tokens)
RWKV_v6_demo.py:70
↓ 1 callersMethoddecodeBytes
(self, tokens)
rwkv_pip_package/src/rwkv/rwkv_tokenizer.py:83
↓ 1 callersMethoddecodeBytes
(self, tokens)
tokenizer/rwkv_tokenizer.py:95
↓ 1 callersMethoddecodeBytes
(self, tokens)
tokenizer/rwkv_tokenizer.py:195
↓ 1 callersMethodencode
(self, string: str, add_special_tokens=False)
run_lm_eval.py:56
↓ 1 callersMethodencode
(self, x)
rwkv_pip_package/src/rwkv/utils.py:46
↓ 1 callersMethodencode
(self, src)
rwkv_pip_package/src/rwkv/rwkv_tokenizer.py:86
↓ 1 callersMethodencode
(self, text)
music/run.py:31
↓ 1 callersMethodencodeBytes
(self, src: bytes)
RWKV_v5_demo.py:49
↓ 1 callersMethodencodeBytes
(self, src: bytes)
RWKV_v6_demo_cuda_bf16.py:215
↓ 1 callersMethodencodeBytes
(self, src: bytes)
RWKV_v6_demo.py:49
↓ 1 callersMethodencodeBytes
(self, src:bytes)
rwkv_pip_package/src/rwkv/rwkv_tokenizer.py:72
↓ 1 callersMethodencodeBytes
(self, src: bytes)
tokenizer/rwkv_tokenizer.py:74
↓ 1 callersMethodencodeBytes
(self, src:bytes)
tokenizer/rwkv_tokenizer.py:169
↓ 1 callersMethodfind_longest
(self, key:bytes, idx:int=0)
rwkv_pip_package/src/rwkv/rwkv_tokenizer.py:35
↓ 1 callersMethodforward
(self, tokens, state, full_output=False)
rwkv_pip_package/src/rwkv/model.py:1443
↓ 1 callersMethodforward_seq
(self, idx:List[int], state:List[torch.Tensor], full_output:bool=False)
rwkv_pip_package/src/rwkv/model.py:426
↓ 1 callersMethodgenerate_zero_state
(self)
rwkv_pip_package/src/rwkv/model.py:284
↓ 1 callersFunctionget_args
()
v2/convert_model.py:12
↓ 1 callersFunctionmatmul_float
(a, b, output_dtype: Optional[torch.dtype]=None)
rwkv_pip_package/src/rwkv/model.py:144
↓ 1 callersFunctionmm8
(x: torch.Tensor, w: torch.Tensor, mx: torch.Tensor, rx: torch.Tensor, my: torch.Tensor, ry: torch.Tensor)
rwkv_pip_package/src/rwkv/model.py:122
↓ 1 callersFunctionmy_qa_generator
(ctx)
API_DEMO_WORLD.py:66
↓ 1 callersFunctionon_message
(message)
chat.py:280
↓ 1 callersFunctionon_message
(message)
v2/chat.py:207
↓ 1 callersMethodprintTokens
(self, tokens)
tokenizer/rwkv_tokenizer.py:104
↓ 1 callersFunctionreply_msg
(msg)
chat.py:277
↓ 1 callersFunctionreply_msg
(msg)
v2/chat.py:204
↓ 1 callersMethodrun_eval
(self, eval_tasks=None, num_fewshot=0, bootstrap_iters=2)
run_lm_eval.py:134
↓ 1 callersFunctionsample_logits
(out, temperature=1.0, top_p=0.8)
RWKV_v5_demo.py:92
↓ 1 callersFunctionsample_logits
(logits, temperature:float=1.0, top_p:float=1.0, top_k:int=0)
RWKV_v6_demo_cuda_bf16.py:154
↓ 1 callersFunctionsample_logits
(out, temperature=1.0, top_p=0.8)
RWKV_in_150_lines.py:118
↓ 1 callersFunctionsample_logits
(out, temperature=1.0, top_p=0.8)
RWKV_v6_demo.py:92
↓ 1 callersMethodsample_logits
(self, logits, temperature=1.0, top_p=0.85, top_k=0)
rwkv_pip_package/src/rwkv/utils.py:55
↓ 1 callersMethodtime_mixing
(self, x, state, i:int, time_mix_k, time_mix_v, time_mix_r, time_mix_g, time_first, time_decay, kw, vw, rw, gw
RWKV_v5_demo.py:169
↓ 1 callersMethodtime_mixing
(self, x, state, i:int, time_mix_k, time_mix_v, time_mix_r, time_first, time_decay, kw, vw, rw, ow)
RWKV_in_150_lines.py:67
↓ 1 callersMethodtime_mixing
(self, x, state, i:int, x_maa, w_maa, k_maa, v_maa, r_maa, g_maa, tm_w1, tm_w2, td_w1, td_w2, time_first, time
RWKV_v6_demo.py:168
↓ 1 callersMethodv5_2_before
(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:1363
↓ 1 callersMethodv6_0_before
(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:1404
MethodFF_one
(self, x, state, i:int, time_mix_k, time_mix_r, kw, vw, rw)
src/model_run.py:137
MethodFF_seq
(self, x, state, i:int, time_mix_k, time_mix_r, kw, vw, rw)
src/model_run.py:149
next →1–100 of 163, ranked by callers