Functions — github.com/HVision-NKU/K-LoRA

CodeHub MCPcopy Create free account

hub / github.com/HVision-NKU/K-LoRA / functions

Functions42 in github.com/HVision-NKU/K-LoRA

⨍ Functions 42 ◇ Types & classes 4

↓ 14 callersFunctioncopy_and_assign_klora_weights

( prefix, attn_module, sub_module_names, lora_weights_content, lora_weights_style, alp

↓ 12 callersFunctioninitialize_klora_layer

( alpha, beta, sum_timesteps, average_ratio, state_dict_1_a, state_dict_1_b, state

↓ 7 callersFunctiontokenize_prompt

(tokenizer, prompt)

train_dreambooth_lora_sdxl.py:871

↓ 6 callersFunctionget_lora_weights

Args: lora_name_or_path (str): huggingface repo id or folder path of lora weights subfolder (Optional[str], optional): sub folder

↓ 4 callersFunctiontext_encoder_lora_state_dict

train_dreambooth_lora_sdxl.py:71

↓ 3 callersFunctioncompute_text_embeddings

(prompt, text_encoders, tokenizers)

train_dreambooth_lora_sdxl.py:1400

↓ 3 callersFunctionget_ratio_between_content_and_style

(lora_weights_content, lora_weights_style)

↓ 2 callersFunctioncompute_time_ids

train_dreambooth_lora_sdxl.py:1383

↓ 2 callersFunctionencode_prompt

(text_encoders, tokenizers, prompt, text_input_ids_list=None)

train_dreambooth_lora_sdxl.py:884

↓ 2 callersFunctionimport_model_class_from_model_name_or_path

( pretrained_model_name_or_path: str, revision: str, subfolder: str = "text_encoder" )

train_dreambooth_lora_sdxl.py:169

↓ 2 callersFunctioninsert_sd_klora_to_unet

( unet, lora_weights_content_path, lora_weights_style_path, alpha, beta, sum_timesteps, pattern )

↓ 2 callersFunctionmerge_community_flux_lora_weights

Args: tensors (torch.Tensor): state dict of lora weights key (str): target attn layer's key prefix (str, optional): prefi

↓ 2 callersFunctionmerge_lora_weights

Args: tensors (torch.Tensor): state dict of lora weights key (str): target attn layer's key prefix (str, optional): prefi

↓ 2 callersFunctionmerge_sd_lora_weights

Args: tensors (torch.Tensor): state dict of lora weights key (str): target attn layer's key prefix (str, optional): prefi

↓ 2 callersFunctionparse_args

inference_gradio.py:9

↓ 2 callersFunctionrename_safetensors_layer_name

↓ 1 callersMethod__init__

( self, in_features: int, out_features: int, weight_1_a: torch.Tensor,

↓ 1 callersFunctioncollate_fn

(examples, with_prior_preservation=False)

train_dreambooth_lora_sdxl.py:837

↓ 1 callersMethodget_klora_weight

(self, timestep)

↓ 1 callersFunctioninsert_community_flux_lora_to_unet

( unet, lora_weights_content_path, lora_weights_style_path, alpha, beta, diffuse_step,

↓ 1 callersFunctionmain

train_dreambooth_lora_sdxl.py:912

↓ 1 callersFunctionparse_args

(input_args=None)

train_dreambooth_lora_sdxl.py:189

↓ 1 callersFunctionparse_args

inference_sd.py:8

↓ 1 callersFunctionparse_args

inference_flux.py:70

↓ 1 callersFunctionrename_key

↓ 1 callersFunctionrun

inference_sd.py:68

↓ 1 callersFunctionrun

inference_flux.py:140

↓ 1 callersFunctionsave_model_card

( repo_id: str, images=None, base_model=str, train_text_encoder=False, instance_prompt=str

train_dreambooth_lora_sdxl.py:103

↓ 1 callersFunctiontext_encoder_attn_modules

train_dreambooth_lora_sdxl.py:74

Method__getitem__

train_dreambooth_lora_sdxl.py:804

Method__getitem__

train_dreambooth_lora_sdxl.py:864

( self, in_features: int, out_features: int, device: Optional[Union[torch.devi

( self, instance_data_root, instance_prompt, class_prompt, class_data_

train_dreambooth_lora_sdxl.py:689

(self, prompt, num_samples)

train_dreambooth_lora_sdxl.py:857

train_dreambooth_lora_sdxl.py:801

train_dreambooth_lora_sdxl.py:861

(self, hidden_states: torch.Tensor)

(self, hidden_states: torch.Tensor)

Functioninsert_community_sd_lora_to_unet

( unet, lora_weights_content_path, lora_weights_style_path, alpha, beta, sum_timesteps )

Functionload_model_hook

(models, input_dir)

train_dreambooth_lora_sdxl.py:1206

inference_gradio.py:58

Functionsave_model_hook

(models, weights, output_dir)

train_dreambooth_lora_sdxl.py:1170