Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Index your code
hub
/
github.com/josStorer/RWKV-Runner
/ types & classes
Types & classes
580 in github.com/josStorer/RWKV-Runner
⨍
Functions
3,079
◇
Types & classes
580
↳
Endpoints
115
↓ 33 callers
Class
RMSNorm
finetune/lora/v6/fla/modules/layernorm.py:554
↓ 28 callers
Class
ShortConvolution
Simple wrapper around `nn.Conv1d` that accepts dimension last.
finetune/lora/v6/fla/modules/convolution.py:79
↓ 27 callers
Class
ChatFormatterResponse
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 callers
Class
BuiltinRule
py310/Lib/site-packages/llama_cpp/llama_grammar.py:459
↓ 10 callers
Class
FakeRequest
backend-python/tests/test_albatross_completion_contract.py:10
↓ 10 callers
Class
FusedCrossEntropyLoss
finetune/lora/v6/fla/modules/fused_cross_entropy.py:323
↓ 9 callers
Class
FakeAlbatross
backend-python/tests/test_albatross_completion_contract.py:39
↓ 8 callers
Class
RuntimeKernelContext
backend-python/albatross/kernel_loader.py:16
↓ 8 callers
Class
suppress_stdout_stderr
py310/Lib/site-packages/llama_cpp/_utils.py:14
↓ 7 callers
Class
ModelConfigBody
backend-python/utils/rwkv.py:27
↓ 6 callers
Class
AlbatrossBackendConfig
backend-python/albatross_engine/config.py:5
↓ 6 callers
Class
FusedRMSNormSwishGate
finetune/lora/v6/fla/modules/fused_norm_gate.py:765
↓ 6 callers
Class
ProfileAccumulator
backend-python/albatross_engine/profiling.py:29
↓ 5 callers
Class
DDLerpLinear
finetune/lora/v6/fla/layers/rwkv6.py:231
↓ 5 callers
Class
ThreadSafeAsyncQueue
桥接线程与 asyncio 的简易队列,确保跨线程 put 安全
backend-python/albatross_engine/core.py:74
↓ 4 callers
Class
AlbatrossRWKV
backend-python/albatross_engine/adapter.py:32
↓ 4 callers
Class
LayerNorm
finetune/lora/v6/fla/modules/layernorm.py:514
↓ 4 callers
Class
RotaryEmbedding
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 callers
Class
ThroughputReporter
backend-python/albatross_engine/throughput.py:26
↓ 3 callers
Class
BlockStateList
finetune/lora/v6/src/infctx_module.py:20
↓ 3 callers
Class
ErrorResponse
OpenAI style error response
py310/Lib/site-packages/llama_cpp/server/errors.py:26
↓ 3 callers
Class
EventLoopQueueDispatcher
Batch cross-thread queue puts into the target event loop.
backend-python/albatross_engine/core.py:32
↓ 3 callers
Class
FakeClock
backend-python/tests/test_albatross_throughput.py:6
↓ 3 callers
Class
HadamardFeatureMap
finetune/lora/v6/fla/modules/feature_map.py:109
↓ 3 callers
Class
HedgehogFeatureMap
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 callers
Class
LerpLinear
finetune/lora/v6/fla/layers/rwkv6.py:194
↓ 3 callers
Class
LlamaChatCompletionHandlerRegistry
py310/Lib/site-packages/llama_cpp/llama_chat_format.py:144
↓ 3 callers
Class
LlamaSamplingContext
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 callers
Class
MMapIndexedDataset
finetune/lora/v6/src/binidx.py:55
↓ 3 callers
Class
MMapIndexedDataset
finetune/lora/v5/src/binidx.py:55
↓ 3 callers
Class
T2RFeatureMap
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 callers
Class
TRIE_TOKENIZER
backend-python/albatross/utils.py:107
↓ 3 callers
Class
Task
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 callers
Class
TrieNode
backend-python/albatross_engine/state_cache.py:43
↓ 2 callers
Class
AbortCriteria
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 callers
Class
AsyncEngineCore
核心引擎类,负责处理核心功能,包括: - Worker 管理 - Task 管理 - 线程与 asyncio 间的安全通信
backend-python/albatross_engine/core.py:98
↓ 2 callers
Class
BlockState
finetune/lora/v6/src/infctx_module.py:14
↓ 2 callers
Class
ChannelMixState
finetune/lora/v6/src/infctx_module.py:9
↓ 2 callers
Class
ChatCompletionRequestMessage
py310/Lib/site-packages/llama_cpp/server/types.py:183
↓ 2 callers
Class
CustomSampler
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 callers
Class
DPFPFeatureMap
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 callers
Class
IndexedDataset
Loader for IndexedDataset
finetune/json2binidx_tool/tools/indexed_dataset.py:133
↓ 2 callers
Class
LinearAttention
finetune/lora/v6/fla/layers/linear_attn.py:14
↓ 2 callers
Class
LlamaSampler
py310/Lib/site-packages/llama_cpp/_internals.py:1817
↓ 2 callers
Class
LlamaSamplingParams
py310/Lib/site-packages/llama_cpp/_internals.py:1188
↓ 2 callers
Class
LlamaTokenizer
py310/Lib/site-packages/llama_cpp/llama_tokenizer.py:45
↓ 2 callers
Class
LoRA
finetune/lora/v6/fla/layers/rwkv6.py:160
↓ 2 callers
Class
LoraLinear
finetune/lora/v6/src/model.py:35
↓ 2 callers
Class
LoraLinear
finetune/lora/v5/src/model.py:161
↓ 2 callers
Class
LoraLinear
finetune/lora/v4/src/model.py:242
↓ 2 callers
Class
MMapIndexedDataset
finetune/lora/v4/src/binidx.py:55
↓ 2 callers
Class
ModelLoadConfig
backend-python/albatross_engine/task.py:182
↓ 2 callers
Class
RWKV
backend-python/rwkv_pip/model.py:1001
↓ 2 callers
Class
RWKV_ChannelMix
finetune/lora/v6/src/model.py:1014
↓ 2 callers
Class
RWKV_ChannelMix
finetune/lora/v5/src/model.py:301
↓ 2 callers
Class
RWKV_ChannelMix
finetune/lora/v4/src/model.py:418
↓ 2 callers
Class
RWKV_x070
backend-python/albatross/rwkv7.py:176
↓ 2 callers
Class
SimpleStateCache
backend-python/albatross_engine/state_cache.py:51
↓ 2 callers
Class
TRIE
backend-python/albatross/utils.py:63
↓ 2 callers
Class
TRIE
backend-python/rwkv_pip/rwkv_tokenizer.py:6
↓ 2 callers
Class
TRIE
finetune/json2binidx_tool/tools/rwkv_tokenizer.py:130
↓ 2 callers
Class
TRIE_TOKENIZER
backend-python/rwkv_pip/rwkv_tokenizer.py:52
↓ 2 callers
Class
TimeMixState
finetune/lora/v6/src/infctx_module.py:3
↓ 2 callers
Class
TrieNode
A node in the prefix tree (Trie).
py310/Lib/site-packages/llama_cpp/llama_cache.py:191
↓ 2 callers
Class
TrieNode
py310/Lib/site-packages/llama_cpp/llama_grammar.py:547
↓ 2 callers
Class
VocabUtils
backend-python/utils/midi.py:142
↓ 2 callers
Class
train_callback
finetune/lora/v6/src/trainer.py:29
↓ 2 callers
Class
train_callback
finetune/lora/v5/src/trainer.py:30
↓ 1 callers
Class
ABCAttention
finetune/lora/v6/fla/layers/abc.py:20
↓ 1 callers
Class
ABCBlock
finetune/lora/v6/fla/models/abc/modeling_abc.py:59
↓ 1 callers
Class
ABCMLP
finetune/lora/v6/fla/models/abc/modeling_abc.py:27
↓ 1 callers
Class
ABCModel
finetune/lora/v6/fla/models/abc/modeling_abc.py:162
↓ 1 callers
Class
ABC_TOKENIZER
backend-python/rwkv_pip/utils.py:38
↓ 1 callers
Class
AlbatrossCompletion
backend-python/albatross_engine/adapter.py:11
↓ 1 callers
Class
AlbatrossProfileAccumulator
backend-python/routes/completion.py:160
↓ 1 callers
Class
AsyncEngineCompletion
异步生成任务的控制器,管理单个生成请求的生命周期
backend-python/albatross_engine/interface.py:27
↓ 1 callers
Class
AugmentValues
backend-python/utils/midi.py:276
↓ 1 callers
Class
BasedLinearAttention
finetune/lora/v6/fla/layers/based.py:17
↓ 1 callers
Class
Block
finetune/lora/v6/src/model.py:1270
↓ 1 callers
Class
Block
finetune/lora/v5/src/model.py:366
↓ 1 callers
Class
Block
finetune/lora/v4/src/model.py:483
↓ 1 callers
Class
CreateChatCompletionRequest
py310/Lib/site-packages/llama_cpp/server/types.py:192
↓ 1 callers
Class
CreateCompletionRequest
py310/Lib/site-packages/llama_cpp/server/types.py:109
↓ 1 callers
Class
CreateEmbeddingRequest
py310/Lib/site-packages/llama_cpp/server/types.py:167
↓ 1 callers
Class
DecodeState
backend-python/utils/midi.py:583
↓ 1 callers
Class
DeltaNet
finetune/lora/v6/fla/layers/delta_net.py:48
↓ 1 callers
Class
DeltaNetBlock
finetune/lora/v6/fla/models/delta_net/modeling_delta_net.py:59
↓ 1 callers
Class
DeltaNetMLP
finetune/lora/v6/fla/models/delta_net/modeling_delta_net.py:27
↓ 1 callers
Class
DeltaNetModel
finetune/lora/v6/fla/models/delta_net/modeling_delta_net.py:162
↓ 1 callers
Class
DetokenizeInputResponse
py310/Lib/site-packages/llama_cpp/server/types.py:311
↓ 1 callers
Class
Encoder
finetune/json2binidx_tool/tools/preprocess_data.py:43
↓ 1 callers
Class
FakeAlbatrossCompletion
backend-python/tests/test_albatross_completion_contract.py:24
↓ 1 callers
Class
FakeAsyncAlbatross
backend-python/tests/test_albatross_completion_contract.py:55
↓ 1 callers
Class
FakeEventLoop
backend-python/tests/test_albatross_core_queue.py:7
↓ 1 callers
Class
FakeTokenizer
backend-python/tests/test_albatross_profiling.py:10
↓ 1 callers
Class
FusedLayerNormSwishGate
finetune/lora/v6/fla/modules/fused_norm_gate.py:724
↓ 1 callers
Class
FusedRMSNormSwishGateLinear
finetune/lora/v6/fla/modules/fused_norm_gate.py:849
↓ 1 callers
Class
GLABlock
finetune/lora/v6/fla/models/gla/modeling_gla.py:59
↓ 1 callers
Class
GLAMLP
finetune/lora/v6/fla/models/gla/modeling_gla.py:27
↓ 1 callers
Class
GLAModel
finetune/lora/v6/fla/models/gla/modeling_gla.py:162
next →
1–100 of 580, ranked by callers