MCPcopy Create free account

hub / github.com/LargeWorldModel/LWM / functions

Functions192 in github.com/LargeWorldModel/LWM

⨍ Functions 192◇ Types & classes 39
↓ 24 callersMethodget_default_config
(updates=None)
lwm/data.py:339
↓ 21 callersMethodencode
(self, pixel_values)
lwm/vqgan.py:52
↓ 8 callersMethoddecode
(self, encoding)
lwm/vqgan.py:55
↓ 8 callersMethodget_jax_mesh
(axis_dims)
lwm/llama.py:202
↓ 6 callersMethodrng_keys
()
lwm/llama.py:297
↓ 5 callersMethodget_partition_rules
Parition rules are orderd, so that the beginning rules match first.
lwm/llama.py:222
↓ 5 callersMethodload_config
(cls, path)
lwm/vqgan.py:101
↓ 4 callersFunctionsave_checkpoint
(train_state, milestone=False)
lwm/train.py:316
↓ 3 callersMethod_split_heads
(self, hidden_states)
lwm/llama.py:434
↓ 2 callersMethod_concatenate_to_cache
(self, key, value, query, attention_mask)
lwm/llama.py:441
↓ 2 callersMethod_process_frame
(self, image, size)
lwm/vision_chat.py:59
↓ 2 callersMethod_wrap_fn
(self, fn)
lwm/vqgan.py:26
↓ 2 callersMethodget_default_config
(cls, updates=None)
lwm/vqgan.py:93
↓ 2 callersMethodjson_iterator
(self)
lwm/data.py:380
↓ 2 callersMethodjson_iterator
(self)
lwm/data.py:588
↓ 2 callersMethodload_dataset
(cls, config, tokenizer, **kwargs)
lwm/data.py:35
↓ 2 callersMethodload_state_dict
(self, state_dict)
lwm/data.py:521
↓ 2 callersMethodparallel_example_iterator
(self)
lwm/data.py:616
↓ 2 callersMethodquantize
(encoding_indices)
lwm/vqgan.py:193
↓ 1 callersMethod__init__
( self, vocab_size=32000, hidden_size=4096, intermediate_size=11008, n
lwm/llama.py:136
↓ 1 callersMethod__init__
(self, vision_vocab_size=8448, tie_vision_embeddings=False, sample_mode='all', **kwargs)
lwm/vision_llama.py:30
↓ 1 callersMethod_decode
(self)
lwm/vqgan.py:43
↓ 1 callersMethod_encode
(self)
lwm/vqgan.py:33
↓ 1 callersMethod_forward_generate
(self)
scripts/eval_needle.py:375
↓ 1 callersMethod_forward_generate
(self)
scripts/eval_needle_multi.py:384
↓ 1 callersMethod_forward_generate
(self)
lwm/vision_chat.py:197
↓ 1 callersMethod_load_model
(self)
scripts/eval_needle.py:327
↓ 1 callersMethod_load_model
(self)
scripts/eval_needle_multi.py:336
↓ 1 callersMethod_load_model
(self)
lwm/vision_chat.py:148
↓ 1 callersMethod_merge_heads
(self, hidden_states)
lwm/llama.py:437
↓ 1 callersMethod_norm
(self, x: jnp.ndarray)
lwm/llama.py:334
↓ 1 callersMethod_read_process_vision
(self, path, max_n_frames)
lwm/vision_chat.py:76
↓ 1 callersMethod_sample_vision
( self, input_ids: None, max_length: Optional[int] = None, pad_token_id: Optio
lwm/vision_llama.py:476
↓ 1 callersFunctionapply_rotary_emb
( xq: jnp.ndarray, xk: jnp.ndarray, freqs_cis: jnp.ndarray, dtype: jnp.dtype=jnp.float32, )
lwm/llama.py:353
↓ 1 callersMethodbatched
(self, iterator, batch_size)
lwm/data.py:398
↓ 1 callersMethodbatched
(self, iterator, batch_size)
lwm/data.py:606
↓ 1 callersMethodcompute_max_input_length
(self, context_length, buffer=1024)
scripts/eval_needle.py:203
↓ 1 callersMethodcompute_max_input_length
(self, context_length, buffer=1024)
scripts/eval_needle_multi.py:207
↓ 1 callersMethodconstruct_input
(self, prompts, max_n_frames)
lwm/vision_chat.py:110
↓ 1 callersMethodcreate_contexts
(self, needle_rnd_number, insert_needle, random_city, trim_context, context_length, depth_percent, seed)
scripts/eval_needle.py:141
↓ 1 callersMethodcreate_contexts
(self, needles_info, random_cities_retrieve, context, context_length, seed)
scripts/eval_needle_multi.py:143
↓ 1 callersMethoddecode
(self, encoding, is_codebook_indices=True)
lwm/vqgan.py:130
↓ 1 callersMethodencode
(self, pixel_values)
lwm/vqgan.py:117
↓ 1 callersMethodgenerate_context
(self, needle, trim_context, context_length, depth_percent)
scripts/eval_needle.py:199
↓ 1 callersMethodgenerate_context
(self, needle, trim_context, context_length, depth_percent)
scripts/eval_needle_multi.py:203
↓ 1 callersFunctiongenerate_first_frame
(prompts, max_input_length)
lwm/vision_generation.py:136
↓ 1 callersMethodgenerate_random_number
(self, num_digits)
scripts/eval_needle.py:109
↓ 1 callersMethodgenerate_random_number
(self, num_digits)
scripts/eval_needle_multi.py:114
↓ 1 callersFunctiongenerate_video_pred
(prompts, images, max_input_length)
lwm/vision_generation.py:189
↓ 1 callersMethodgenerate_vision
( self, input_ids: jnp.ndarray, cfg_scales: jnp.ndarray, generation_config: Op
lwm/vision_llama.py:583
↓ 1 callersMethodget_default_config
(updates=None)
lwm/data.py:58
↓ 1 callersMethodget_default_config
(updates=None)
lwm/data.py:128
↓ 1 callersMethodget_default_config
(updates=None)
lwm/data.py:248
↓ 1 callersMethodget_default_config
(updates=None)
lwm/data.py:547
↓ 1 callersMethodget_ranks_and_size
(mesh)
lwm/llama.py:206
↓ 1 callersMethodget_state_dict
(self)
lwm/data.py:513
↓ 1 callersMethodget_weight_decay_exclusions
()
lwm/llama.py:286
↓ 1 callersMethodinit_cache
r""" Args: batch_size (`int`): batch_size used for fast auto-regressive decoding. Defines the batch size of the in
lwm/llama.py:806
↓ 1 callersMethodinit_cache
(self, batch_size, max_length)
lwm/vision_llama.py:143
↓ 1 callersMethodinsert_needle
(self, needle, context, depth_percent, context_length)
scripts/eval_needle.py:162
↓ 1 callersMethodinsert_needle
(self, needle, context, depth_percent, context_length)
scripts/eval_needle_multi.py:166
↓ 1 callersMethodlogistic
(self, x, L=100, x0=50, k=.1)
scripts/eval_needle.py:114
↓ 1 callersMethodlogistic
(self, x, L=100, x0=50, k=.1)
scripts/eval_needle_multi.py:119
↓ 1 callersMethodparallel_example_iterator
(self)
lwm/data.py:408
↓ 1 callersMethodparse_json
(self, line)
lwm/data.py:370
↓ 1 callersMethodparse_json
(self, line)
lwm/data.py:578
↓ 1 callersFunctionprecompute_freqs_cis
(dim: int, max_position_embedding: int, theta: float=10000.0, dtype: jnp.dtype=jnp.float32)
lwm/llama.py:344
↓ 1 callersMethodprepare_inputs_for_generation
( self, input_ids, max_length, attention_mask: Optional[jax.Array] = None, vision_masks = None )
lwm/vision_llama.py:447
↓ 1 callersMethodprint_start_test_summary
(self)
scripts/eval_needle.py:295
↓ 1 callersMethodprint_start_test_summary
(self)
scripts/eval_needle_multi.py:304
↓ 1 callersMethodread_context_files
(self, n)
scripts/eval_needle.py:121
↓ 1 callersMethodread_context_files
(self, n)
scripts/eval_needle_multi.py:126
↓ 1 callersMethodrun_test
(self)
scripts/eval_needle.py:209
↓ 1 callersMethodrun_test
(self)
scripts/eval_needle_multi.py:214
↓ 1 callersMethodstart_test
(self)
scripts/eval_needle.py:303
↓ 1 callersMethodstart_test
(self)
scripts/eval_needle_multi.py:312
↓ 1 callersMethodtext_processor
(self)
lwm/data.py:321
↓ 1 callersMethodtext_processor
(self)
lwm/data.py:537
↓ 1 callersMethodtext_processor
(self)
lwm/data.py:837
↓ 1 callersMethodupdate_inputs_for_generation
(self, model_outputs, model_kwargs)
lwm/vision_llama.py:468
Method__call__
(self, prompts, max_input_length)
scripts/eval_needle.py:402
Method__call__
(self, prompts, max_input_length)
scripts/eval_needle_multi.py:411
Method__call__
(self, prompts, max_n_frames)
lwm/vision_chat.py:222
Method__call__
(self, pixel_values)
lwm/vqgan.py:143
Method__call__
(self, pixel_values)
lwm/vqgan.py:153
Method__call__
(self, hidden_states)
lwm/vqgan.py:171
Method__call__
(self, z, encoding_indices=None)
lwm/vqgan.py:192
Method__call__
(self, hidden_states)
lwm/vqgan.py:229
Method__call__
(self, hidden_states)
lwm/vqgan.py:248
Method__call__
(self, hidden_states)
lwm/vqgan.py:268
Method__call__
(self, hidden_states)
lwm/vqgan.py:290
Method__call__
(self, hidden_states)
lwm/vqgan.py:310
Method__call__
(self, hidden_states)
lwm/vqgan.py:327
Method__call__
(self, hidden_states)
lwm/vqgan.py:346
Method__call__
(self, x: jnp.ndarray)
lwm/llama.py:337
Method__call__
( self, hidden_states, attention_mask, segment_ids, position_ids,
lwm/llama.py:494
Method__call__
(self, x: jnp.ndarray, deterministic: bool = True)
lwm/llama.py:658
Method__call__
( self, hidden_states, attention_mask=None, segment_ids=None, position
lwm/llama.py:704
Method__call__
( self, input_ids, attention_mask=None, segment_ids=None, position_ids
lwm/llama.py:827
Method__call__
( self, hidden_states, attention_mask=None, segment_ids=None, position
lwm/llama.py:905
next →1–100 of 192, ranked by callers