MCPcopy Index your code

hub / github.com/josStorer/RWKV-Runner / types & classes

Types & classes580 in github.com/josStorer/RWKV-Runner

↓ 33 callersClassRMSNorm
finetune/lora/v6/fla/modules/layernorm.py:554
↓ 28 callersClassShortConvolution
Simple wrapper around `nn.Conv1d` that accepts dimension last.
finetune/lora/v6/fla/modules/convolution.py:79
↓ 27 callersClassChatFormatterResponse
Dataclass that stores completion parameters for a given chat format and create_chat_completion request. prompt contains the formatted prompt
py310/Lib/site-packages/llama_cpp/llama_chat_format.py:195
↓ 18 callersClassBuiltinRule
py310/Lib/site-packages/llama_cpp/llama_grammar.py:459
↓ 10 callersClassFakeRequest
backend-python/tests/test_albatross_completion_contract.py:10
↓ 10 callersClassFusedCrossEntropyLoss
finetune/lora/v6/fla/modules/fused_cross_entropy.py:323
↓ 9 callersClassFakeAlbatross
backend-python/tests/test_albatross_completion_contract.py:39
↓ 8 callersClassRuntimeKernelContext
backend-python/albatross/kernel_loader.py:16
↓ 8 callersClasssuppress_stdout_stderr
py310/Lib/site-packages/llama_cpp/_utils.py:14
↓ 7 callersClassModelConfigBody
backend-python/utils/rwkv.py:27
↓ 6 callersClassAlbatrossBackendConfig
backend-python/albatross_engine/config.py:5
↓ 6 callersClassFusedRMSNormSwishGate
finetune/lora/v6/fla/modules/fused_norm_gate.py:765
↓ 6 callersClassProfileAccumulator
backend-python/albatross_engine/profiling.py:29
↓ 5 callersClassDDLerpLinear
finetune/lora/v6/fla/layers/rwkv6.py:231
↓ 5 callersClassThreadSafeAsyncQueue
桥接线程与 asyncio 的简易队列,确保跨线程 put 安全
backend-python/albatross_engine/core.py:74
↓ 4 callersClassAlbatrossRWKV
backend-python/albatross_engine/adapter.py:32
↓ 4 callersClassLayerNorm
finetune/lora/v6/fla/modules/layernorm.py:514
↓ 4 callersClassRotaryEmbedding
The rotary position embeddings from RoFormer_ (Su et. al). A crucial insight from the method is that the query and keys are transformed b
finetune/lora/v6/fla/modules/rotary.py:137
↓ 4 callersClassThroughputReporter
backend-python/albatross_engine/throughput.py:26
↓ 3 callersClassBlockStateList
finetune/lora/v6/src/infctx_module.py:20
↓ 3 callersClassErrorResponse
OpenAI style error response
py310/Lib/site-packages/llama_cpp/server/errors.py:26
↓ 3 callersClassEventLoopQueueDispatcher
Batch cross-thread queue puts into the target event loop.
backend-python/albatross_engine/core.py:32
↓ 3 callersClassFakeClock
backend-python/tests/test_albatross_throughput.py:6
↓ 3 callersClassHadamardFeatureMap
finetune/lora/v6/fla/modules/feature_map.py:109
↓ 3 callersClassHedgehogFeatureMap
r""" Hedgehog feature map as introduced in `The Hedgehog & the Porcupine: Expressive Linear Attentions with Softmax Mimicry <https://arxiv.org
finetune/lora/v6/fla/modules/feature_map.py:37
↓ 3 callersClassLerpLinear
finetune/lora/v6/fla/layers/rwkv6.py:194
↓ 3 callersClassLlamaChatCompletionHandlerRegistry
py310/Lib/site-packages/llama_cpp/llama_chat_format.py:144
↓ 3 callersClassLlamaSamplingContext
High-level Python wrapper that manages the lifecycle and configuration of the llama.cpp sampler chain.
py310/Lib/site-packages/llama_cpp/_internals.py:1330
↓ 3 callersClassMMapIndexedDataset
finetune/lora/v6/src/binidx.py:55
↓ 3 callersClassMMapIndexedDataset
finetune/lora/v5/src/binidx.py:55
↓ 3 callersClassT2RFeatureMap
r""" Simple linear mapping feature map as in `Finetuning Pretrained Transformers into RNNs <https://arxiv.org/abs/2103.13076>`_
finetune/lora/v6/fla/modules/feature_map.py:65
↓ 3 callersClassTRIE_TOKENIZER
backend-python/albatross/utils.py:107
↓ 3 callersClassTask
Represents a text generation task with associated configuration and state. This class encapsulates all necessary information for managing a singl
backend-python/albatross_engine/task.py:93
↓ 3 callersClassTrieNode
backend-python/albatross_engine/state_cache.py:43
↓ 2 callersClassAbortCriteria
Listen for external interruption signals to trigger a stop condition. When an external thread calls `llama.abort()`, a loop interrupt is gene
py310/Lib/site-packages/llama_cpp/llama.py:65
↓ 2 callersClassAsyncEngineCore
核心引擎类,负责处理核心功能,包括: - Worker 管理 - Task 管理 - 线程与 asyncio 间的安全通信
backend-python/albatross_engine/core.py:98
↓ 2 callersClassBlockState
finetune/lora/v6/src/infctx_module.py:14
↓ 2 callersClassChannelMixState
finetune/lora/v6/src/infctx_module.py:9
↓ 2 callersClassChatCompletionRequestMessage
py310/Lib/site-packages/llama_cpp/server/types.py:183
↓ 2 callersClassCustomSampler
Python wrapper for llama.cpp custom sampler. apply_func: Callable receiving llama_token_data_array and modifying logits in-p
py310/Lib/site-packages/llama_cpp/_internals.py:1701
↓ 2 callersClassDPFPFeatureMap
r""" Deterministic Parameter-Free Projection (DPFP) feature map in `Linear Transformers Are Secretly Fast Weight Programmers <https://arxiv.or
finetune/lora/v6/fla/modules/feature_map.py:87
↓ 2 callersClassIndexedDataset
Loader for IndexedDataset
finetune/json2binidx_tool/tools/indexed_dataset.py:133
↓ 2 callersClassLinearAttention
finetune/lora/v6/fla/layers/linear_attn.py:14
↓ 2 callersClassLlamaSampler
py310/Lib/site-packages/llama_cpp/_internals.py:1817
↓ 2 callersClassLlamaSamplingParams
py310/Lib/site-packages/llama_cpp/_internals.py:1188
↓ 2 callersClassLlamaTokenizer
py310/Lib/site-packages/llama_cpp/llama_tokenizer.py:45
↓ 2 callersClassLoRA
finetune/lora/v6/fla/layers/rwkv6.py:160
↓ 2 callersClassLoraLinear
finetune/lora/v6/src/model.py:35
↓ 2 callersClassLoraLinear
finetune/lora/v5/src/model.py:161
↓ 2 callersClassLoraLinear
finetune/lora/v4/src/model.py:242
↓ 2 callersClassMMapIndexedDataset
finetune/lora/v4/src/binidx.py:55
↓ 2 callersClassModelLoadConfig
backend-python/albatross_engine/task.py:182
↓ 2 callersClassRWKV
backend-python/rwkv_pip/model.py:1001
↓ 2 callersClassRWKV_ChannelMix
finetune/lora/v6/src/model.py:1014
↓ 2 callersClassRWKV_ChannelMix
finetune/lora/v5/src/model.py:301
↓ 2 callersClassRWKV_ChannelMix
finetune/lora/v4/src/model.py:418
↓ 2 callersClassRWKV_x070
backend-python/albatross/rwkv7.py:176
↓ 2 callersClassSimpleStateCache
backend-python/albatross_engine/state_cache.py:51
↓ 2 callersClassTRIE
backend-python/albatross/utils.py:63
↓ 2 callersClassTRIE
backend-python/rwkv_pip/rwkv_tokenizer.py:6
↓ 2 callersClassTRIE
finetune/json2binidx_tool/tools/rwkv_tokenizer.py:130
↓ 2 callersClassTRIE_TOKENIZER
backend-python/rwkv_pip/rwkv_tokenizer.py:52
↓ 2 callersClassTimeMixState
finetune/lora/v6/src/infctx_module.py:3
↓ 2 callersClassTrieNode
A node in the prefix tree (Trie).
py310/Lib/site-packages/llama_cpp/llama_cache.py:191
↓ 2 callersClassTrieNode
py310/Lib/site-packages/llama_cpp/llama_grammar.py:547
↓ 2 callersClassVocabUtils
backend-python/utils/midi.py:142
↓ 2 callersClasstrain_callback
finetune/lora/v6/src/trainer.py:29
↓ 2 callersClasstrain_callback
finetune/lora/v5/src/trainer.py:30
↓ 1 callersClassABCAttention
finetune/lora/v6/fla/layers/abc.py:20
↓ 1 callersClassABCBlock
finetune/lora/v6/fla/models/abc/modeling_abc.py:59
↓ 1 callersClassABCMLP
finetune/lora/v6/fla/models/abc/modeling_abc.py:27
↓ 1 callersClassABCModel
finetune/lora/v6/fla/models/abc/modeling_abc.py:162
↓ 1 callersClassABC_TOKENIZER
backend-python/rwkv_pip/utils.py:38
↓ 1 callersClassAlbatrossCompletion
backend-python/albatross_engine/adapter.py:11
↓ 1 callersClassAlbatrossProfileAccumulator
backend-python/routes/completion.py:160
↓ 1 callersClassAsyncEngineCompletion
异步生成任务的控制器,管理单个生成请求的生命周期
backend-python/albatross_engine/interface.py:27
↓ 1 callersClassAugmentValues
backend-python/utils/midi.py:276
↓ 1 callersClassBasedLinearAttention
finetune/lora/v6/fla/layers/based.py:17
↓ 1 callersClassBlock
finetune/lora/v6/src/model.py:1270
↓ 1 callersClassBlock
finetune/lora/v5/src/model.py:366
↓ 1 callersClassBlock
finetune/lora/v4/src/model.py:483
↓ 1 callersClassCreateChatCompletionRequest
py310/Lib/site-packages/llama_cpp/server/types.py:192
↓ 1 callersClassCreateCompletionRequest
py310/Lib/site-packages/llama_cpp/server/types.py:109
↓ 1 callersClassCreateEmbeddingRequest
py310/Lib/site-packages/llama_cpp/server/types.py:167
↓ 1 callersClassDecodeState
backend-python/utils/midi.py:583
↓ 1 callersClassDeltaNet
finetune/lora/v6/fla/layers/delta_net.py:48
↓ 1 callersClassDeltaNetBlock
finetune/lora/v6/fla/models/delta_net/modeling_delta_net.py:59
↓ 1 callersClassDeltaNetMLP
finetune/lora/v6/fla/models/delta_net/modeling_delta_net.py:27
↓ 1 callersClassDeltaNetModel
finetune/lora/v6/fla/models/delta_net/modeling_delta_net.py:162
↓ 1 callersClassDetokenizeInputResponse
py310/Lib/site-packages/llama_cpp/server/types.py:311
↓ 1 callersClassEncoder
finetune/json2binidx_tool/tools/preprocess_data.py:43
↓ 1 callersClassFakeAlbatrossCompletion
backend-python/tests/test_albatross_completion_contract.py:24
↓ 1 callersClassFakeAsyncAlbatross
backend-python/tests/test_albatross_completion_contract.py:55
↓ 1 callersClassFakeEventLoop
backend-python/tests/test_albatross_core_queue.py:7
↓ 1 callersClassFakeTokenizer
backend-python/tests/test_albatross_profiling.py:10
↓ 1 callersClassFusedLayerNormSwishGate
finetune/lora/v6/fla/modules/fused_norm_gate.py:724
↓ 1 callersClassFusedRMSNormSwishGateLinear
finetune/lora/v6/fla/modules/fused_norm_gate.py:849
↓ 1 callersClassGLABlock
finetune/lora/v6/fla/models/gla/modeling_gla.py:59
↓ 1 callersClassGLAMLP
finetune/lora/v6/fla/models/gla/modeling_gla.py:27
↓ 1 callersClassGLAModel
finetune/lora/v6/fla/models/gla/modeling_gla.py:162
next →1–100 of 580, ranked by callers