MCPcopy Create free account

hub / github.com/OpenMOSS/MOSS / functions

Functions130 in github.com/OpenMOSS/MOSS

⨍ Functions 130◇ Types & classes 21↳ Endpoints 3
Methodexecute
( self, input_ids: Optional[jt.Var] = None, past_key_values: Optional[Tuple[Tuple[jt.V
models_jittor/model.py:255
Methodexecute
( self, input_ids: Optional[jt.Var] = None, past_key_values: Optional[Tuple[Tuple[jt.V
models_jittor/model.py:358
Methodfind_re
(string, pattern, start_pos)
models/tokenization_moss.py:355
Methodforward
( self, hidden_states: Optional[torch.FloatTensor], layer_past: Optional[Tuple[torch.T
models/modeling_moss.py:151
Methodforward
(self, hidden_states: Optional[torch.FloatTensor])
models/modeling_moss.py:241
Methodforward
( self, hidden_states: Optional[torch.FloatTensor], layer_past: Optional[Tuple[torch.T
models/modeling_moss.py:258
Methodforward
( self, input_ids: Optional[torch.LongTensor] = None, past_key_values: Optional[Tuple[
models/modeling_moss.py:422
Methodforward
r""" labels (`torch.LongTensor` of shape `(batch_size, sequence_length)`, *optional*): Labels for language modeling. Note that the
models/modeling_moss.py:655
Methodforward
(ctx, input, qweight, scales, qzeros, g_idx, bits, maxq)
models/quantization.py:278
Methodforward
(self, x)
models/quantization.py:365
Functiongenerate_answer
()
moss_web_demo_streamlit.py:92
Methodget_input_embeddings
(self)
models/modeling_moss.py:410
Methodget_output_embeddings
(self)
models/modeling_moss.py:616
Methodget_vocab
(self)
models/tokenization_moss.py:193
Methodkernel_call
()
models/custom_autotune.py:63
Functionmatmul_248_kernel
Compute the matrix multiplication C = A x B. A is of shape (M, K) float16 B is of shape (K//8, N) int32 C is of shape (M, N) float16
models/quantization.py:57
Methodnoop
(*args, **kwargs)
models/modeling_moss.py:598
Methodpack
(self, linear, scales, zeros, g_idx=None)
models/quantization.py:315
Functionpostprocess
(self, y)
moss_web_demo_gradio.py:68
Functionpredict
(input, chatbot, max_length, top_p, temperature, history)
moss_web_demo_gradio.py:115
Methodprepare_for_tokenization
(self, text, is_split_into_words=False, **kwargs)
models/tokenization_moss.py:304
Methodprepare_inputs_for_generation
(self, input_ids, past_key_values=None, **kwargs)
models/modeling_moss.py:622
Functionreset_state
()
moss_web_demo_gradio.py:150
Functionreset_user_input
()
moss_web_demo_gradio.py:146
Methodsave_vocabulary
(self, save_directory: str, filename_prefix: Optional[str] = None)
models/tokenization_moss.py:275
Methodset_input_embeddings
(self, new_embeddings)
models/modeling_moss.py:413
Methodset_output_embeddings
(self, new_embeddings)
models/modeling_moss.py:619
Functiontrans_matmul_248_kernel
Compute the matrix multiplication C = A x B. A is of shape (M, N) float16 B is of shape (K//8, N) int32 C is of shape (M, K) float16
models/quantization.py:168
Methodvocab_size
(self)
models/tokenization_moss.py:190
Methodwarmup
(self, *args, **kwargs)
models/custom_autotune.py:121
← previous101–130 of 130, ranked by callers