Functions — github.com/adobe-research/custom-diffusion

CodeHub MCPcopy Create free account

hub / github.com/adobe-research/custom-diffusion / functions

Functions135 in github.com/adobe-research/custom-diffusion

⨍ Functions 135◇ Types & classes 19

customconcept101/evaluate.py:42

customconcept101/evaluate.py:68

src/diffusers_data_pipeline.py:265

src/diffusers_data_pipeline.py:333

src/finetune_data.py:174

customconcept101/evaluate.py:33

customconcept101/evaluate.py:63

customconcept101/evaluate.py:88

Method_predict_dataloader

(self, shuffle=False)

Method_test_dataloader

(self, shuffle=False)

Method_testtube

(self, pl_module, images, batch_idx, split)

Method_train_dataloader

Method_val_dataloader

(self, shuffle=False)

Methodadd_token

(self, initializer_token)

src/diffusers_model_pipeline.py:423

Methodchange_checkpoint

src/model.py:135

Methodchange_forward

src/model.py:168

Methodconfigure_optimizers

src/model.py:178

(*args, **kwargs)

src/custom_modules.py:314

src/custom_modules.py:293

Methodload_model

(self, save_path, compress=False)

src/diffusers_model_pipeline.py:605

Methodnew_forward

(self, x, context=None, mask=None)

src/model.py:144

Methodon_keyboard_interrupt

(self, trainer, pl_module)

Methodon_pretrain_routine_start

(self, trainer, pl_module)

Methodon_train_batch_end

(self, trainer, pl_module, outputs, batch, batch_idx, dataloader_idx)

Methodon_train_epoch_end

(self, trainer, pl_module, outputs)

Methodon_train_epoch_start

(self, trainer, pl_module)

(self, x_start, cond, t, mask=None, noise=None)

src/model.py:221

Methodsave_pretrained

(self, save_path, freeze_model="crossattn_kv", save_text_encoder=False, all=False)

src/diffusers_model_pipeline.py:576

Functionset_use_memory_efficient_attention_xformers

( self, use_memory_efficient_attention_xformers: bool, attention_op: Optional[Callable] = None )

src/diffusers_model_pipeline.py:236

Functionstr2bool

Methodtraining_step

(self, batch, batch_idx)

src/model.py:275

Functionworker_init_fn

← previous101–135 of 135, ranked by callers