MCPcopy Create free account

hub / github.com/LargeWorldModel/LWM / types & classes

Types & classes39 in github.com/LargeWorldModel/LWM

↓ 4 callersClassLLaMAConfig
lwm/llama.py:133
↓ 4 callersClassRMSNorm
lwm/llama.py:320
↓ 4 callersClassResnetBlock
lwm/vqgan.py:242
↓ 4 callersClassVideoLLaMAConfig
lwm/vision_llama.py:27
↓ 3 callersClassAttnBlock
lwm/vqgan.py:266
↓ 2 callersClassFlaxLLaMABlockCollection
lwm/llama.py:898
↓ 2 callersClassFlaxLLaMAForCausalLM
lwm/llama.py:1110
↓ 2 callersClassFlaxVideoLLaMAForCausalLM
lwm/vision_llama.py:444
↓ 2 callersClassMidBlock
lwm/vqgan.py:340
↓ 2 callersClassTextProcessor
Example processor that converts a dictionary of texts into tokens.
lwm/data.py:55
↓ 2 callersClassVQGAN
lwm/vqgan.py:14
↓ 1 callersClassDecoder
lwm/vqgan.py:167
↓ 1 callersClassDownsample
lwm/vqgan.py:286
↓ 1 callersClassDownsamplingBlock
lwm/vqgan.py:224
↓ 1 callersClassEncoder
lwm/vqgan.py:149
↓ 1 callersClassFlaxLLaMAModule
lwm/llama.py:982
↓ 1 callersClassFlaxVideoLLaMAModule
lwm/vision_llama.py:255
↓ 1 callersClassHuggingfaceDataset
Huggingface dataset, where the dataset is loaded using the huggingface datasets.load_dataset() function.
lwm/data.py:242
↓ 1 callersClassJsonDataset
JSON dataset, where each line of the data file contains a JSON dictionary with text fields.
lwm/data.py:333
↓ 1 callersClassJsonVisionDataset
lwm/data.py:545
↓ 1 callersClassLLMNeedleHaystackTester
scripts/eval_needle.py:47
↓ 1 callersClassLLMNeedleHaystackTester
scripts/eval_needle_multi.py:50
↓ 1 callersClassSampler
scripts/eval_needle.py:310
↓ 1 callersClassSampler
scripts/eval_needle_multi.py:319
↓ 1 callersClassSampler
lwm/vision_chat.py:40
↓ 1 callersClassUpsample
lwm/vqgan.py:306
↓ 1 callersClassUpsamplingBlock
lwm/vqgan.py:322
↓ 1 callersClassVQGANModel
lwm/vqgan.py:105
↓ 1 callersClassVectorQuantizer
lwm/vqgan.py:187
↓ 1 callersClassVisionTextProcessor
lwm/data.py:126
ClassDatasetFactory
Datset builder class.
lwm/data.py:16
ClassFlaxLLaMAAttention
lwm/llama.py:378
ClassFlaxLLaMABlock
lwm/llama.py:664
ClassFlaxLLaMAForCausalLMModule
lwm/llama.py:1049
ClassFlaxLLaMAMLP
lwm/llama.py:623
ClassFlaxLLaMAPreTrainedModel
An abstract class to handle weights initialization and a simple interface for downloading and loading pretrained models.
lwm/llama.py:747
ClassFlaxVideoLLaMAForCausalLMModule
lwm/vision_llama.py:346
ClassFlaxVideoLLaMAPreTrainedModel
An abstract class to handle weights initialization and a simple interface for downloading and loading pretrained models.
lwm/vision_llama.py:121
ClassVQGANConfig
lwm/vqgan.py:59