MCPcopy Create free account

hub / github.com/Jellyfish042/RWKV_Othello / functions

Functions154 in github.com/Jellyfish042/RWKV_Othello

Method__init__
(self)
othello.py:5
Method__init__
(self, player1, player2, opening_book=None)
arena.py:8
Method__init__
(self, print_to_console=True)
logger.py:2
Method__init__
(self, engine_class, engine_path: str, level: int, threads: int, pool_size: int, save_path: str)
generate_data.py:13
Method__init__
(self, engine_class, engine_path, level, threads)
generate_data.py:55
Method__init__
(self, master)
ui.py:11
Method__init__
(self, model_path=MODEL_PATH)
rwkv_for_ui.py:8
Method__init__
(self, path, level=5, threads=2)
alphabeta_engine.py:8
Method__init__
(self, path, level=15, threads=2)
engine.py:7
Function__nop
(ob)
rwkv_extended.py:24
Method_ai_think_thread
(self, search_config)
ui.py:898
Method_execute_ai_move
(self, move: str)
ui.py:854
Method_generate_sample
(self, input_moves: str, max_width: int, max_depth: int)
generate_data.py:31
Method_process_ai_update
(self, data: Dict[str, Any])
ui.py:871
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_extended.py:889
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_extended.py:946
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_extended.py:1012
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_extended.py:1118
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_extended.py:913
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_extended.py:971
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_extended.py:1039
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_extended.py:1082
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_extended.py:1156
Methodcallback_wrapper
(self, model_output: str)
rwkv_for_ui.py:31
Methodclear_counts
(self)
rwkv_engine.py:41
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_extended.py:1201
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_extended.py:1247
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_extended.py:1286
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_extended.py:839
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_extended.py:862
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_extended.py:850
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_extended.py:874
Functionformat_node
(node)
formatter.py:114
Methodforward
(ctx, state, r, w, k, v, a, b)
rwkv_extended.py:200
Methodforward
(self, idx, state, full_output=False)
rwkv_extended.py:270
Methodforward
(ctx, B, T, C, H, state, r, k, v, w, u)
rwkv_extended.py:767
Methodforward
(ctx, B, T, C, H, state, r, k, v, w, u)
rwkv_extended.py:800
Methodforward
(self, tokens, state, full_output=False)
rwkv_extended.py:1298
Functionforward_bf16
cuda/rwkv7_op.cpp:12
Functionforward_fp16
cuda/rwkv7_op.cpp:15
Functionforward_fp32
cuda/rwkv7_op.cpp:18
Methodhas_valid_moves
(self, player)
othello.py:52
Methodlog_func
(self, text: str, end="\n")
logger.py:6
Functionmm8_one
cuda/wrapper.cpp:84
Functionmm8_seq
cuda/wrapper.cpp:50
Methodon_model_change
Reinitialize AI when model changes
ui.py:936
Methodprint_all
(self, max_length=2000)
logger.py:11
Methodprint_board
(self)
othello.py:141
Methodprint_board
(self)
engine.py:114
Methodset_state_by_board
(self, board)
engine.py:93
Functionsort_key
(item)
formatter.py:135
Methodtransform_coordinates
(move_count, eval_value)
ui.py:355
Methodupdate_callback
(data: Dict[str, Any])
ui.py:900
Functionwkv_forward
cuda/wrapper.cpp:27
← previous101–154 of 154, ranked by callers