MCPcopy Create free account

hub / github.com/TinyLLaVA/TinyLLaVA_Factory / functions

Functions354 in github.com/TinyLLaVA/TinyLLaVA_Factory

Functionload_demo
()
tinyllava/serve/app.py:83
Functionload_image_from_base64
(image)
tinyllava/utils/eval_utils.py:49
Methodload_model
(self, **kwargs)
tinyllava/model/connector/qformer.py:63
Methodload_model
(self, vision_tower_name, **kwargs)
tinyllava/model/vision_tower/base.py:23
Functionlora_kbit_setting
(model, training_args)
tinyllava/utils/train_utils.py:14
Functionmake_inputs_require_grad
(module, input, output)
tinyllava/utils/train_utils.py:10
Methodmake_inputs_require_grad
(module, input, output)
tinyllava/training_recipe/base.py:158
Methodmake_labels
(self, input_ids, prompt, tokenizer)
tinyllava/data/template/pretrain_template.py:25
Methodmessages
(self)
tinyllava/utils/message.py:30
Methodmodality_lengths
(self)
tinyllava/data/dataset.py:49
Methodoutput_hook
(module, input, output)
tinyllava_visualizer/tinyllava_visualizer.py:137
Methodprepare_inputs_for_generation
(self, input_ids, past_key_values=None, inputs_embeds=None, **kwargs)
tinyllava/model/modeling_tinyllava.py:205
Methodprepare_inputs_for_generation
( self, input_ids, past_key_values=None, attention_mask=None, inputs_e
tinyllava/model/llm/openelm.py:1200
Methodprepare_inputs_for_generation
( self, input_ids, query_embeds, past=None, attention_mask=None, **model_kwargs )
tinyllava/model/connector/qformer.py:1163
Functionregenerate
(state, image_process_mode)
tinyllava/serve/app.py:53
Functionregister_connector
(name)
tinyllava/model/connector/__init__.py:17
Functionregister_connector_cls
(cls)
tinyllava/model/connector/__init__.py:18
Functionregister_llm
(name)
tinyllava/model/llm/__init__.py:17
Functionregister_llm_cls
(cls)
tinyllava/model/llm/__init__.py:18
Functionregister_template
(name)
tinyllava/data/template/__init__.py:16
Functionregister_template_cls
(cls)
tinyllava/data/template/__init__.py:17
Functionregister_training_recipe
(name)
tinyllava/training_recipe/__init__.py:17
Functionregister_training_recipe_cls
(cls)
tinyllava/training_recipe/__init__.py:18
Functionregister_vision_tower
(name)
tinyllava/model/vision_tower/__init__.py:18
Functionregister_vision_tower_cls
(cls)
tinyllava/model/vision_tower/__init__.py:19
Methodresize_token_embeddings
(self, new_num_tokens: Optional[int] = None, pad_to_multiple_of=None)
tinyllava/model/modeling_tinyllava.py:99
Functionreturn_gemmaclass
()
tinyllava/model/llm/gemma.py:6
Functionreturn_openelmclass
()
tinyllava/model/llm/openelm.py:35
Functionreturn_phiclass
()
tinyllava/model/llm/stablelm.py:6
Functionreturn_phiclass
()
tinyllava/model/llm/phi.py:6
Functionreturn_qwen2class
()
tinyllava/model/llm/qwen2.py:6
Functionreturn_tinyllamaclass
()
tinyllava/model/llm/tinyllama.py:6
Methodsave
(self, model, trainer)
tinyllava/training_recipe/qlora_recipe.py:70
Methodsave
(self, model, trainer)
tinyllava/training_recipe/lora_recipe.py:52
Methodsave_attn_gradients
(self, attn_gradients)
tinyllava/model/connector/qformer.py:215
Methodset_decoder
(self, decoder)
tinyllava/model/modeling_tinyllava.py:90
Methodset_decoder
(self, decoder)
tinyllava/model/llm/openelm.py:1119
Methodset_input_embeddings
(self, value)
tinyllava/model/modeling_tinyllava.py:81
Methodset_input_embeddings
(self, new_embeddings: torch.Tensor)
tinyllava/model/llm/openelm.py:874
Methodset_input_embeddings
(self, value)
tinyllava/model/llm/openelm.py:1110
Methodset_input_embeddings
(self, value)
tinyllava/model/connector/qformer.py:768
Methodset_output_embeddings
(self, new_embeddings)
tinyllava/model/modeling_tinyllava.py:87
Methodset_output_embeddings
(self, new_embeddings)
tinyllava/model/llm/openelm.py:1116
Methodset_output_embeddings
(self, new_embeddings)
tinyllava/model/connector/qformer.py:1050
Methodset_output_embeddings
(self, new_embeddings)
tinyllava/model/connector/qformer.py:1213
Methodtie_weights
(self)
tinyllava/model/modeling_tinyllava.py:96
Functiontokenizer_and_post_load
(tokenizer)
tinyllava/model/llm/qwen2.py:7
Functiontokenizer_and_post_load
(tokenizer)
tinyllava/model/llm/tinyllama.py:7
Functiontokenizer_and_post_load
(tokenizer)
tinyllava/model/llm/openelm.py:36
Functiontokenizer_and_post_load
(tokenizer)
tinyllava/model/llm/gemma.py:7
Functiontokenizer_and_post_load
(tokenizer)
tinyllava/model/llm/stablelm.py:7
Functiontokenizer_and_post_load
(tokenizer)
tinyllava/model/llm/phi.py:7
Methodtraining_model_converse
(self, model)
tinyllava/training_recipe/qlora_recipe.py:45
Methodtraining_model_converse
(self, model)
tinyllava/training_recipe/lora_recipe.py:26
← previous301–354 of 354, ranked by callers