MCPcopy Create free account

hub / github.com/Yuanshi9815/OminiControl / functions

Functions52 in github.com/Yuanshi9815/OminiControl

↓ 10 callersFunctionlora_forward
Apply a single, explicitly-selected LoRA adapter to ``module`` at scale 1.0. This is a fast, allocation-free replacement for the previous
omini/pipeline/flux_omini.py:144
↓ 5 callersFunctionget_config
()
omini/train_flux/trainer.py:26
↓ 5 callersFunctiontrain
(dataset, trainable_model, config, test_function)
omini/train_flux/trainer.py:338
↓ 4 callersFunctionconvert_to_condition
( condition_type: str, raw_img: Union[Image.Image, torch.Tensor], blur_radius: Optional[int] = 5,
omini/pipeline/flux_omini.py:69
↓ 4 callersFunctiongenerate
( pipeline: FluxPipeline, prompt: Union[str, List[str]] = None, prompt_2: Optional[Union[str, List
omini/pipeline/flux_omini.py:556
↓ 3 callersFunctionclip_hidden_states
(hidden_states: torch.FloatTensor)
omini/pipeline/flux_omini.py:31
↓ 3 callersMethodencode
( self, pipe: FluxPipeline, empty: bool = False )
omini/pipeline/flux_omini.py:122
↓ 3 callersFunctionencode_images
Encodes the images into tokens and ids for FLUX pipeline.
omini/pipeline/flux_omini.py:37
↓ 3 callersFunctiontransformer_forward
( transformer: FluxTransformer2DModel, image_features: List[torch.Tensor], text_features: List[tor
omini/pipeline/flux_omini.py:456
↓ 2 callersMethod__get_condition__
(self, image, condition_type)
omini/train_flux/train_spatial_alignment.py:42
↓ 2 callersFunction_is_cond
(idx: int)
omini/pipeline/flux_omini.py:294
↓ 2 callersFunction_stash
(canvas, tokens, ids, H, W)
omini/pipeline/flux_omini.py:819
↓ 2 callersFunctionattn_forward
( attn: Attention, hidden_states: List[torch.FloatTensor], adapters: List[str], hidden_states2
omini/pipeline/flux_omini.py:214
↓ 2 callersFunctionget_rank
()
omini/train_flux/trainer.py:18
↓ 1 callersFunction_adanorm_zero_forward
AdaLayerNormZero.forward with LoRA applied to the inner ``linear``.
omini/pipeline/flux_omini.py:187
↓ 1 callersFunction_adanorm_zero_single_forward
AdaLayerNormZeroSingle.forward with LoRA applied to the inner ``linear``.
omini/pipeline/flux_omini.py:195
↓ 1 callersFunction_feedforward_forward
FeedForward.forward routing each sub-module through ``lora_forward``. Only the output projection (``ff.net[2]``) is LoRA-wrapped in practice;
omini/pipeline/flux_omini.py:203
↓ 1 callersFunctionblock_forward
( self, image_hidden_states: List[torch.FloatTensor], text_hidden_states: List[torch.FloatTensor],
omini/pipeline/flux_omini.py:361
↓ 1 callersFunctionclear_cache
()
omini/pipeline/flux_omini.py:703
↓ 1 callersFunctionget_temb
(timestep, guidance, pooled_projection)
omini/pipeline/flux_omini.py:493
↓ 1 callersMethodinit_lora
(self, lora_path: str, lora_config: dict)
omini/train_flux/trainer.py:86
↓ 1 callersFunctioninit_wandb
(wandb_config, run_name)
omini/train_flux/trainer.py:34
↓ 1 callersFunctionmain
()
omini/train_flux/train_token_integration.py:96
↓ 1 callersFunctionmain
()
omini/train_flux/train_custom.py:26
↓ 1 callersFunctionmain
()
omini/train_flux/train_spatial_alignment.py:170
↓ 1 callersFunctionmain
()
omini/train_flux/train_subject.py:151
↓ 1 callersFunctionmain
()
omini/train_flux/train_multi_condition.py:117
↓ 1 callersMethodsave_lora
(self, path: str)
omini/train_flux/trainer.py:102
↓ 1 callersFunctionsingle_block_forward
( self, hidden_states: List[torch.FloatTensor], tembs: List[torch.FloatTensor], adapters: List
omini/pipeline/flux_omini.py:418
Method__getitem__
(self, idx)
omini/train_flux/train_token_integration.py:15
Method__getitem__
(self, idx)
omini/train_flux/train_custom.py:15
Method__getitem__
(self, idx)
omini/train_flux/train_spatial_alignment.py:75
Method__getitem__
(self, idx)
omini/train_flux/train_subject.py:44
Method__getitem__
(self, idx)
omini/train_flux/train_multi_condition.py:15
Method__init__
( self, base_dataset, condition_size=(512, 512), target_size=(512, 512),
omini/train_flux/train_spatial_alignment.py:17
Method__init__
( self, base_dataset, condition_size=(512, 512), target_size=(512, 512),
omini/train_flux/train_subject.py:17
Method__init__
( self, flux_pipe_id: str, lora_path: str = None, lora_config: dict = None,
omini/train_flux/trainer.py:49
Method__init__
(self, run_name, training_config: dict = {}, test_function=None)
omini/train_flux/trainer.py:259
Method__init__
( self, condition: Union[Image.Image, torch.Tensor], adapter_setting: Union[str, dict]
omini/pipeline/flux_omini.py:104
Method__len__
(self)
omini/train_flux/train_spatial_alignment.py:39
Method__len__
(self)
omini/train_flux/train_subject.py:41
Methodconfigure_optimizers
(self)
omini/train_flux/trainer.py:113
Functionfilter_func
(item)
omini/train_flux/train_subject.py:161
Methodgenerate_a_sample
(self)
omini/train_flux/trainer.py:254
Methodon_train_batch_end
(self, trainer, pl_module, outputs, batch, batch_idx)
omini/train_flux/trainer.py:275
Functionseed_everything
(seed: int = 42)
omini/pipeline/flux_omini.py:25
Functiontest_function
(model, save_path, file_name)
omini/train_flux/train_token_integration.py:55
Functiontest_function
(model, save_path, file_name)
omini/train_flux/train_custom.py:21
Functiontest_function
(model, save_path, file_name)
omini/train_flux/train_spatial_alignment.py:108
Functiontest_function
(model, save_path, file_name)
omini/train_flux/train_subject.py:105
Functiontest_function
(model, save_path, file_name)
omini/train_flux/train_multi_condition.py:57
Methodtraining_step
(self, batch, batch_idx)
omini/train_flux/trainer.py:139