Methodchat(self, tokenizer, query: str, history: List[Tuple[str, str]] = None, max_length: int = 2048, num_beams=1,
models/chatglm/modeling_chatglm.py:1215
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)
models/chatrwkv/rwkv_pip_package/src/rwkv/model.py:366
Methodffn_one(self, x, sx, ln_w, ln_b, k_mix, r_mix, kw, vw, rw)
models/chatrwkv/rwkv_pip_package/src/rwkv/model.py:277
Methodffn_seq(self, x, sx, ln_w, ln_b, k_mix, r_mix, kw, vw, rw)
models/chatrwkv/rwkv_pip_package/src/rwkv/model.py:288
Methodforward(self, input_ids, position_ids, attention_mask,
tokentype_ids=None, layer_past=None, get_key_v
models/pangualpha/megatron/model/language_model.py:438
Methodforward(self, input_ids, position_ids, attention_mask, labels=None,
tokentype_ids=None, layer_past=No
models/pangualpha/megatron/model/gpt2_model.py:54
Methodforward(self, hidden_states, query_hidden_state, attention_mask, layer_past=None,
get_key_value=False
models/pangualpha/megatron/model/transformer.py:413
Methodforward(self, hidden_states, query_hidden_state, attention_mask, layer_past=None,
get_key_value=False
models/pangualpha/megatron/model/transformer.py:737
Methodforward(self, hidden_states, query_hidden_state, attention_mask, layer_past=None,
get_key_value=False
models/pangualpha/megatron/model/transformer.py:891