MCPcopy Create free account

hub / github.com/Jittor/JittorLLMs / functions

Functions1,370 in github.com/Jittor/JittorLLMs

Functionbuild_prompt
(history)
models/chatglm/__init__.py:11
Functionbuild_prompt
(history)
models/atom7b/__init__.py:29
Functionbuild_sample_idx
models/pangualpha/megatron/data/helpers.cpp:36
Functionbwd
models/pangualpha/megatron/fused_kernels/scaled_upper_triang_masked_softmax.cpp:42
Methodcan_save_slow_tokenizer
`bool`: Whether or not the slow tokenizer can be saved. Usually for sentencepiece based slow tokenizer, this can only be `True` if th
models/qwen2/qwen2_jt/tokenization_utils_fast.py:108
Functioncb
()
models/pangualpha/megatron/mpu/random.py:82
Methodchat
(self)
models/pangualpha/__init__.py:145
Methodchat
(self)
models/llama/__init__.py:102
Methodchat
(self, tokenizer, query: str, history: List[Tuple[str, str]] = None, max_length: int = 2048, num_beams=1,
models/chatglm/modeling_chatglm.py:1215
Methodchat
(self)
models/chatrwkv/__init__.py:322
Functioncheckpoint
Checkpoint a model or part of the model. This has been directly copied from torch.utils.checkpoint.
models/pangualpha/megatron/mpu/random.py:329
Methodclean_up_tokenization
Clean up a list of simple English tokenization artifacts like spaces before punctuations and abbreviated forms. Args: ou
models/qwen2/qwen2_jt/tokenization_utils_fast.py:177
Functionclip_grad_norm
Clips gradient norm of an iterable of parameters. This is adapted from torch.nn.utils.clip_grad.clip_grad_norm_ and added functionality to ha
models/pangualpha/megatron/mpu/grads.py:74
Methodcls
(self)
models/pangualpha/megatron/tokenizer/tokenizer.py:103
Methodcls
(self)
models/pangualpha/megatron/tokenizer/tokenizer.py:174
Methodcommand_token_vocab
dictionary mapping command tokens to ids for tokenizer
models/pangualpha/megatron/deprecated_data_utils/tokenization.py:305
Methodcommand_tokens
list (or iterable) of all command tokens for tokenizer
models/pangualpha/megatron/deprecated_data_utils/tokenization.py:300
Functioncompare
(a)
models/pangualpha/test/compare_tensor_mindspore_gpu.py:3
Functionconfigure_data
add cmdline flags for configuring datasets
models/pangualpha/megatron/deprecated_data_utils/configure_data.py:235
Methodconstruct
(self, x)
models/pangualpha/testLayerNorm.py:17
Functionconvert_ids_to_tokens
(inv_vocab, ids)
models/pangualpha/megatron/tokenizer/bert_tokenization.py:148
Methodconvert_ids_to_tokens
Converts a sequence of ids in BPE tokens using the vocab.
models/pangualpha/megatron/deprecated_data_utils/tokenization_gpt2.py:267
Methodconvert_ids_to_tokens
(self, ids)
models/pangualpha/megatron/tokenizer/bert_tokenization.py:181
Functionconvert_tokens_to_ids
(vocab, tokens)
models/pangualpha/megatron/tokenizer/bert_tokenization.py:144
Methodconvert_tokens_to_ids
(self, tokens)
models/pangualpha/megatron/tokenizer/bert_tokenization.py:178
Methodconvert_tokens_to_ids
(self, tokens)
models/pangualpha/megatron/tokenizer/tokenization_jieba.py:78
Methodcreate_custom_forward
(module)
models/atom7b/modeling_llama.py:703
Functioncreate_doc_idx
(sizes)
models/pangualpha/megatron/data/indexed_dataset.py:119
Functioncreate_item
(request: Request)
api.py:14
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
Methodcummulative_sizes
(self)
models/pangualpha/megatron/deprecated_data_utils/datasets.py:117
Methodcustom
(start, end)
models/pangualpha/megatron/model/transformer.py:871
Methodcustom_forward
(*inputs)
models/pangualpha/megatron/model/transformer.py:872
Methodcustom_forward
(*inputs)
models/atom7b/modeling_llama.py:704
Functiondataset_exists
(path, impl)
models/pangualpha/megatron/data/indexed_dataset.py:75
Methoddecode
( self, token_ids: Union[List[int], List[List[int]]], skip_special_tokens:
models/chatglm/tokenization_chatglm.py:259
Methoddecode_token_ids
(self, token_ids)
models/pangualpha/megatron/tokenizer/tokenizer.py:159
Methoddecoder
`tokenizers.decoders.Decoder`: The Rust decoder for this tokenizer.
models/qwen2/qwen2_jt/tokenization_utils_fast.py:170
Functiondestroy_model_parallel
Set the groups to none.
models/pangualpha/megatron/mpu/initialize.py:157
Methoddetokenize
(self, token_ids)
models/pangualpha/megatron/tokenizer/tokenizer.py:216
Methoddetokenize
(self, token_ids)
models/pangualpha/megatron/tokenizer/tokenization_jieba.py:84
Functiondispatch_scaled_upper_triang_masked_softmax_backward
models/pangualpha/megatron/fused_kernels/scaled_upper_triang_masked_softmax.h:352
Functiondispatch_scaled_upper_triang_masked_softmax_forward
models/pangualpha/megatron/fused_kernels/scaled_upper_triang_masked_softmax.h:263
Methoddoc_idx
(self)
models/pangualpha/megatron/data/indexed_dataset.py:435
Methoddoc_idx
(self)
models/pangualpha/megatron/data/indexed_dataset.py:519
Methodencode
@param text: Text to encode. @param linebreak: Whether to encode newline (\n) in text. @param whitespaces: Whether to encode
models/chatglm/tokenization_chatglm.py:114
Methodend_document
(self)
models/pangualpha/megatron/data/indexed_dataset.py:551
Functionensure_directory_exists
Build filename's path if it does not already exists.
models/pangualpha/tools/splitMergedCkpt_v0.py:43
Methodeod
(self)
models/pangualpha/megatron/tokenizer/tokenizer.py:118
Methodeod
(self)
models/pangualpha/megatron/tokenizer/tokenizer.py:220
Methodeod
(self)
models/pangualpha/megatron/tokenizer/tokenization_jieba.py:69
Methodeop_token_id
`Optional[int]`: Id of the end of sentence token in the vocabulary. Returns `None` if the token has not been set.
models/chatglm/tokenization_chatglm.py:220
Functionerf_gelu
(x)
models/pangualpha/megatron/model/utils.py:63
Methodexecute
(self, x)
models/qwen2/qwen2_jt/modeling_qwen2.py:132
Methodexecute
( self, hidden_states: jt.Var, position_embeddings: Tuple[jt.Var, jt.Var], att
models/qwen2/qwen2_jt/modeling_qwen2.py:187
Methodexecute
(self, hidden_states)
models/qwen2/qwen2_jt/modeling_qwen2.py:241
Methodexecute
( self, hidden_states: jt.Var, attention_mask: Optional[jt.Var] = None, positi
models/qwen2/qwen2_jt/modeling_qwen2.py:261
Methodexecute
(self, x, position_ids)
models/qwen2/qwen2_jt/modeling_qwen2.py:341
Methodexecute
( self, input_ids: jt.Var = None, attention_mask: Optional[jt.Var] = None, pos
models/qwen2/qwen2_jt/modeling_qwen2.py:425
Methodexecute
( self, input_ids: jt.Var = None, attention_mask: Optional[jt.Var] = None, pos
models/qwen2/qwen2_jt/modeling_qwen2.py:500
Methodexecute
(self, x)
models/llama/llama/model.py:37
Methodexecute
(self, x: jt.Var, start_pos: int, freqs_cis: jt.Var, mask: Optional[jt.Var])
models/llama/llama/model.py:110
Methodexecute
(self, x)
models/llama/llama/model.py:166
Methodexecute
(self, x: jt.Var, start_pos: int, freqs_cis: jt.Var, mask: Optional[jt.Var])
models/llama/llama/model.py:184
Methodexecute
(self, tokens: jt.Var, start_pos: int)
models/llama/llama/model.py:214
Methodexists
(path)
models/pangualpha/megatron/data/indexed_dataset.py:533
Methodexists
(model_path)
models/pangualpha/megatron/deprecated_data_utils/tokenization.py:491
Methodexists
(model_path)
models/pangualpha/megatron/deprecated_data_utils/tokenization.py:629
Methodextra_repr
(self)
models/qwen2/qwen2_jt/modeling_qwen2.py:248
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
Functionfilename_to_url
Return the url and etag (which may be ``None``) stored for `filename`. Raise ``EnvironmentError`` if `filename` or its stored metadata do not
models/pangualpha/megatron/deprecated_data_utils/file_utils.py:61
Methodfinalize
(self, index_file)
models/pangualpha/megatron/data/indexed_dataset.py:566
Functionfloat_conversion
(val)
models/pangualpha/megatron/fp16/fp16.py:58
Functionforward
models/chatrwkv/rwkv_pip_package/src/rwkv/cuda/wkv_op.cpp:5
Methodforward
(ctx, input_)
models/pangualpha/pretrain_ict.py:49
Methodforward
(self, input_ids, attention_mask, tokentype_ids)
models/pangualpha/megatron/model/classification.py:53
Methodforward
(self, input_ids, attention_mask, tokentype_ids=None)
models/pangualpha/megatron/model/realm_model.py:169
Methodforward
(ctx, inputs, scale)
models/pangualpha/megatron/model/fused_softmax.py:26
Methodforward
(self, *inputs, **kwargs)
models/pangualpha/megatron/model/distributed.py:74
Methodforward
(self, hidden_states, sequence_index=0)
models/pangualpha/megatron/model/language_model.py:85
Methodforward
(self, input_ids, position_ids, tokentype_ids=None)
models/pangualpha/megatron/model/language_model.py:166
Methodforward
(self, position_ids, tokentype_ids=None)
models/pangualpha/megatron/model/language_model.py:312
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
(ctx, input, bias)
models/pangualpha/megatron/model/fused_bias_gelu.py:50
Methodforward
(self, hidden_states)
models/pangualpha/megatron/model/transformer.py:94
Methodforward
(self, hidden_states, attention_mask, layer_past=None, get_key_value=False)
models/pangualpha/megatron/model/transformer.py:184
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, attention_mask, layer_past=None, get_key_value=False)
models/pangualpha/megatron/model/transformer.py:629
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
Methodforward
(self, hidden_states, word_embeddings_weight)
models/pangualpha/megatron/model/bert_model.py:101
Methodforward
(self, input_ids, attention_mask, tokentype_ids=None, lm_labels=None)
models/pangualpha/megatron/model/bert_model.py:143
Methodforward
(self, input_ids, attention_mask, tokentype_ids)
models/pangualpha/megatron/model/multiple_choice.py:51
Methodforward
(self, *inputs, **kwargs)
models/pangualpha/megatron/fp16/fp16.py:73
Methodforward
(self, input)
models/pangualpha/megatron/fp16/fp16util.py:38
Methodforward
(self, *inputs)
models/pangualpha/megatron/fp16/fp16util.py:100
Methodforward
(self, input_)
models/pangualpha/megatron/mpu/layers.py:170
Methodforward
(self, input_)
models/pangualpha/megatron/mpu/layers.py:272
← previousnext →1,001–1,100 of 1,370, ranked by callers