Functions — github.com/YaYaB/finetune-diffusion

CodeHub MCPcopy Create free account

hub / github.com/YaYaB/finetune-diffusion / functions

Functions12 in github.com/YaYaB/finetune-diffusion

⨍ Functions 12 ◇ Types & classes 1

↓ 3 callersMethodstep

(self, new_model)

train_text_to_image.py:272

↓ 1 callersMethodget_decay

Compute the decay factor for the exponential moving average.

train_text_to_image.py:264

↓ 1 callersFunctionget_full_repo_name

(model_id: str, organization: Optional[str] = None, token: Optional[str] = None)

train_text_to_image.py:228

↓ 1 callersFunctionget_paramters

create_dataset.py:10

↓ 1 callersFunctionload_blip_model

↓ 1 callersFunctionload_image

(image_size, device, path_image=None)

↓ 1 callersFunctionmain

train_text_to_image.py:301

↓ 1 callersFunctionparse_args

train_text_to_image.py:31

↓ 1 callersFunctiontokenize_captions

(examples, is_train=True)

train_text_to_image.py:441

( self, model, decay=0.9999, device=None, )

train_text_to_image.py:248

Functioncollate_fn

train_text_to_image.py:480

Functionpreprocess_train

train_text_to_image.py:467