Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/LargeWorldModel/LWM
/ functions
Functions
192 in github.com/LargeWorldModel/LWM
⨍
Functions
192
◇
Types & classes
39
Method
__call__
( self, input_ids, attention_mask, segment_ids, position_ids,
lwm/llama.py:1002
Method
__call__
( self, input_ids, attention_mask=None, segment_ids=None, position_ids
lwm/llama.py:1066
Method
__call__
(self, example, has_aux=False, add_bos_token=True, add_eos_token=True)
lwm/data.py:77
Method
__call__
(self, example, has_aux=False, add_bos_token=True, add_eos_token=True)
lwm/data.py:153
Method
__call__
( self, input_ids, vision_masks, attention_mask=None, segment_ids=None
lwm/vision_llama.py:179
Method
__call__
( self, input_ids, vision_masks, attention_mask, segment_ids,
lwm/vision_llama.py:283
Method
__call__
( self, input_ids, vision_masks, attention_mask=None, segment_ids=None
lwm/vision_llama.py:371
Method
__init__
(self, needle="", haystack_file="", retrieval_question="Wha
scripts/eval_needle.py:64
Method
__init__
(self)
scripts/eval_needle.py:311
Method
__init__
(self, needle="", haystack_file="", retrieval_question="Wha
scripts/eval_needle_multi.py:67
Method
__init__
(self)
scripts/eval_needle_multi.py:320
Method
__init__
(self)
lwm/vision_chat.py:41
Method
__init__
(self, vqgan_checkpoint, replicate=False)
lwm/vqgan.py:15
Method
__init__
( self, resolution=256, num_channels=3, hidden_channels=128, channel_m
lwm/vqgan.py:62
Method
__init__
( self, config: LLaMAConfig, input_shape: Tuple = (1, 1), seed: int = 0,
lwm/llama.py:757
Method
__init__
(self)
lwm/data.py:51
Method
__init__
(self, config, tokenizer)
lwm/data.py:70
Method
__init__
(self, config, tokenizer)
lwm/data.py:144
Method
__init__
(self, config, tokenizer, text_processor)
lwm/data.py:262
Method
__init__
(self, config, tokenizer, text_processor, node_info)
lwm/data.py:360
Method
__init__
(self, config, tokenizer, text_processor, node_info)
lwm/data.py:568
Method
__init__
( self, config: VideoLLaMAConfig, input_shape: Tuple = (4, 1), seed: int = 0,
lwm/vision_llama.py:131
Method
__iter__
(self)
lwm/data.py:272
Method
__iter__
(self)
lwm/data.py:434
Method
__iter__
(self)
lwm/data.py:642
Function
_forward_generate
(params, rng, batch, n_tokens, cfg_scale, top_k, temperature)
lwm/vision_generation.py:106
Method
_iter_no_pad
(self)
lwm/data.py:736
Method
_iter_pad
(self)
lwm/data.py:651
Method
_key_from_result
(result)
scripts/eval_needle.py:214
Method
_key_from_result
(result)
scripts/eval_needle_multi.py:219
Method
_make_callback
(self, v)
lwm/data.py:510
Method
_make_callback
(self, v)
lwm/data.py:810
Method
block_size
(self)
scripts/eval_needle.py:319
Method
block_size
(self)
scripts/eval_needle_multi.py:328
Method
block_size
(self)
lwm/vision_chat.py:52
Function
create_trainstate_from_params
(params)
lwm/train.py:141
Method
data_dim
(self)
scripts/eval_needle.py:324
Method
data_dim
(self)
scripts/eval_needle_multi.py:333
Method
data_dim
(self)
lwm/vision_chat.py:56
Method
dataset
(self)
lwm/data.py:325
Method
encode_and_trim
(self, context, context_length)
scripts/eval_needle.py:135
Method
encode_and_trim
(self, context, context_length)
scripts/eval_needle_multi.py:137
Function
eval_step
(train_state, rng, batch)
lwm/train.py:225
Method
exists
(fname)
scripts/eval_needle.py:219
Method
exists
(fname)
scripts/eval_needle_multi.py:224
Method
fn
(params, rng, batch)
scripts/eval_needle.py:376
Method
fn
(params, rng, batch)
scripts/eval_needle_multi.py:385
Method
fn
(params, rng, batch)
lwm/vision_chat.py:198
Method
fn
(pixel_values, params)
lwm/vqgan.py:34
Method
fn
(cached_key, cached_value, key, value, cur_index)
lwm/llama.py:454
Method
get_default_config
(cls, updates=None)
lwm/llama.py:193
Method
get_default_config
(updates=None)
lwm/data.py:20
Method
get_frozen_param_exclusions
(freeze_base)
lwm/llama.py:290
Method
get_partition_rules
Parition rules are orderd, so that the beginning rules match first.
lwm/vision_llama.py:37
Method
get_state_dict
(self)
lwm/data.py:305
Method
get_state_dict
(self)
lwm/data.py:813
Function
init_fn
(rng)
lwm/train.py:144
Method
init_weights
(self, rng: jax.random.PRNGKey, input_shape: Tuple, params: FrozenDict = None)
lwm/llama.py:769
Method
init_weights
(self, rng, input_shape, params=None)
lwm/vision_llama.py:156
Method
load_config
(cls, path)
lwm/llama.py:301
Method
load_config
(cls, path)
lwm/vision_llama.py:107
Method
load_state_dict
(self, state_dict)
lwm/data.py:308
Method
load_state_dict
(self, state_dict)
lwm/data.py:821
Function
loss_and_accuracy
(params)
lwm/train.py:169
Function
main
(argv)
scripts/eval_needle.py:427
Function
main
(argv)
scripts/eval_needle_multi.py:436
Function
main
(argv)
lwm/vision_chat.py:236
Function
main
(argv)
lwm/train.py:59
Function
main
(argv)
lwm/vision_generation.py:44
Method
prepare_inputs_for_generation
( self, input_ids, max_length, attention_mask: Optional[jax.Array] = None, )
lwm/llama.py:1113
Method
sample_search_body_fn
state update fn.
lwm/vision_llama.py:534
Method
sample_search_cond_fn
state termination condition fn.
lwm/vision_llama.py:527
Method
seq_length
(self)
lwm/data.py:313
Method
seq_length
(self)
lwm/data.py:529
Method
seq_length
(self)
lwm/data.py:829
Method
setup
(self)
lwm/vqgan.py:108
Method
setup
(self)
lwm/llama.py:326
Method
setup
(self)
lwm/llama.py:384
Method
setup
(self)
lwm/llama.py:629
Method
setup
(self)
lwm/llama.py:670
Method
setup
(self)
lwm/llama.py:988
Method
setup
(self)
lwm/llama.py:1055
Method
setup
(self)
lwm/vision_llama.py:261
Method
setup
(self)
lwm/vision_llama.py:352
Method
tokenizer
(self)
lwm/data.py:317
Method
tokenizer
(self)
lwm/data.py:533
Method
tokenizer
(self)
lwm/data.py:833
Function
train_step
(train_state, rng, batch)
lwm/train.py:166
Method
update_inputs_for_generation
(self, model_outputs, model_kwargs)
lwm/llama.py:1134
Method
vocab_size
(self)
lwm/data.py:329
Method
vocab_size
(self)
lwm/data.py:541
Method
vocab_size
(self)
lwm/data.py:841
← previous
101–192 of 192, ranked by callers