Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/LargeWorldModel/LWM
/ types & classes
Types & classes
39 in github.com/LargeWorldModel/LWM
⨍
Functions
192
◇
Types & classes
39
↓ 4 callers
Class
LLaMAConfig
lwm/llama.py:133
↓ 4 callers
Class
RMSNorm
lwm/llama.py:320
↓ 4 callers
Class
ResnetBlock
lwm/vqgan.py:242
↓ 4 callers
Class
VideoLLaMAConfig
lwm/vision_llama.py:27
↓ 3 callers
Class
AttnBlock
lwm/vqgan.py:266
↓ 2 callers
Class
FlaxLLaMABlockCollection
lwm/llama.py:898
↓ 2 callers
Class
FlaxLLaMAForCausalLM
lwm/llama.py:1110
↓ 2 callers
Class
FlaxVideoLLaMAForCausalLM
lwm/vision_llama.py:444
↓ 2 callers
Class
MidBlock
lwm/vqgan.py:340
↓ 2 callers
Class
TextProcessor
Example processor that converts a dictionary of texts into tokens.
lwm/data.py:55
↓ 2 callers
Class
VQGAN
lwm/vqgan.py:14
↓ 1 callers
Class
Decoder
lwm/vqgan.py:167
↓ 1 callers
Class
Downsample
lwm/vqgan.py:286
↓ 1 callers
Class
DownsamplingBlock
lwm/vqgan.py:224
↓ 1 callers
Class
Encoder
lwm/vqgan.py:149
↓ 1 callers
Class
FlaxLLaMAModule
lwm/llama.py:982
↓ 1 callers
Class
FlaxVideoLLaMAModule
lwm/vision_llama.py:255
↓ 1 callers
Class
HuggingfaceDataset
Huggingface dataset, where the dataset is loaded using the huggingface datasets.load_dataset() function.
lwm/data.py:242
↓ 1 callers
Class
JsonDataset
JSON dataset, where each line of the data file contains a JSON dictionary with text fields.
lwm/data.py:333
↓ 1 callers
Class
JsonVisionDataset
lwm/data.py:545
↓ 1 callers
Class
LLMNeedleHaystackTester
scripts/eval_needle.py:47
↓ 1 callers
Class
LLMNeedleHaystackTester
scripts/eval_needle_multi.py:50
↓ 1 callers
Class
Sampler
scripts/eval_needle.py:310
↓ 1 callers
Class
Sampler
scripts/eval_needle_multi.py:319
↓ 1 callers
Class
Sampler
lwm/vision_chat.py:40
↓ 1 callers
Class
Upsample
lwm/vqgan.py:306
↓ 1 callers
Class
UpsamplingBlock
lwm/vqgan.py:322
↓ 1 callers
Class
VQGANModel
lwm/vqgan.py:105
↓ 1 callers
Class
VectorQuantizer
lwm/vqgan.py:187
↓ 1 callers
Class
VisionTextProcessor
lwm/data.py:126
Class
DatasetFactory
Datset builder class.
lwm/data.py:16
Class
FlaxLLaMAAttention
lwm/llama.py:378
Class
FlaxLLaMABlock
lwm/llama.py:664
Class
FlaxLLaMAForCausalLMModule
lwm/llama.py:1049
Class
FlaxLLaMAMLP
lwm/llama.py:623
Class
FlaxLLaMAPreTrainedModel
An abstract class to handle weights initialization and a simple interface for downloading and loading pretrained models.
lwm/llama.py:747
Class
FlaxVideoLLaMAForCausalLMModule
lwm/vision_llama.py:346
Class
FlaxVideoLLaMAPreTrainedModel
An abstract class to handle weights initialization and a simple interface for downloading and loading pretrained models.
lwm/vision_llama.py:121
Class
VQGANConfig
lwm/vqgan.py:59