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

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

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

⨍ Functions 135◇ Types & classes 19

↓ 6 callersFunctionextract_all_images

(images, model, datasetclass, device, batch_size=64, num_workers=8)

customconcept101/evaluate.py:105

↓ 5 callersMethodencode

(self, *args, **kwargs)

src/custom_modules.py:222

↓ 5 callersMethodsave_pretrained

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

src/diffusers_model_pipeline.py:450

↓ 5 callersFunctiontokenize_prompt

(tokenizer, prompt)

src/diffusers_training_sdxl.py:329

↓ 3 callersFunctionisimage

(path)

src/finetune_data.py:110

↓ 3 callersFunctionpreprocess

(image, scale, resample)

src/diffusers_data_pipeline.py:226

↓ 3 callersMethodshared_step

(self, batch, **kwargs)

src/model.py:298

↓ 2 callersMethod__init__

(self, batch_frequency, max_images, clamp=True, increase_log_steps=True, rescale=True, disabl

train.py:493

↓ 2 callersFunctioncollate_fn

(examples, with_prior_preservation)

src/diffusers_data_pipeline.py:233

↓ 2 callersFunctioncompute_text_embeddings

(prompt, text_encoders, tokenizers)

src/diffusers_training_sdxl.py:1073

↓ 2 callersFunctioncompute_time_ids

()

src/diffusers_training_sdxl.py:1059

↓ 2 callersFunctionencode_prompt

(text_encoders, tokenizers, prompt, text_input_ids_list=None)

src/diffusers_training_sdxl.py:301

↓ 2 callersFunctionget_text_embedding

(prompts)

src/composenW.py:92

↓ 2 callersFunctionget_text_embedding

(prompts)

src/diffusers_composenW.py:55

↓ 2 callersFunctionimport_model_class_from_model_name_or_path

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

src/diffusers_training_sdxl.py:382

↓ 2 callersMethodload_model

(self, save_path, compress=False)

src/diffusers_model_pipeline.py:474

↓ 1 callersMethod__init__

( self, vae: AutoencoderKL, text_encoder: CLIPTextModel, tokenizer: CLIPTokeni

src/diffusers_model_pipeline.py:397

↓ 1 callersMethod__init__

(self)

src/custom_modules.py:219

↓ 1 callersMethod_transform_test

(self, n_px)

customconcept101/evaluate.py:47

↓ 1 callersMethod_transform_test

(self, n_px)

customconcept101/evaluate.py:73

↓ 1 callersMethodadd_token

(self)

src/custom_modules.py:243

↓ 1 callersFunctioncalmetrics

(sample_root, target_paths, numgen, outpkl)

customconcept101/evaluate.py:224

↓ 1 callersFunctionchange_attn

(unet)

src/diffusers_training.py:274

↓ 1 callersFunctionchange_attn

(unet)

src/diffusers_training_sdxl.py:287

↓ 1 callersMethodcheck_frequency

(self, check_idx)

train.py:576

↓ 1 callersFunctionclipeval

(image_dir, candidates_json, device)

customconcept101/evaluate.py:152

↓ 1 callersFunctionclipeval_image

(image_dir, image_dir_ref, device)

customconcept101/evaluate.py:178

↓ 1 callersFunctioncompose

(paths, category, outpath, pretrained_model_path, regularization_prompt, prompts, save_path, device='cuda')

src/composenW.py:68

↓ 1 callersFunctioncompose

(paths, category, outpath, pretrained_model_path, regularization_prompt, prompts, save_path, device='cuda')

src/diffusers_composenW.py:44

↓ 1 callersFunctioncompress

(delta_ckpt, ckpt, diffuser=False, compression_ratio=0.6, device='cuda')

src/compress.py:8

↓ 1 callersFunctionconvert

(ckpt, delta_ckpt, sd_version, config, modelname, mode)

src/convert.py:51

↓ 1 callersFunctioncreate_custom_diffusion

(unet, freeze_model)

src/diffusers_training.py:254

↓ 1 callersFunctioncreate_custom_diffusion

(unet, freeze_model)

src/diffusers_training_sdxl.py:267

↓ 1 callersMethodcustom_forward

r""" Returns:

src/custom_modules.py:259

↓ 1 callersFunctiondinoeval_image

(image_dir, image_dir_ref, device)

customconcept101/evaluate.py:201

↓ 1 callersFunctionextract_all_captions

(captions, model, device, batch_size=256, num_workers=8, append=False)

customconcept101/evaluate.py:92

↓ 1 callersMethodfreeze

(self)

src/custom_modules.py:284

↓ 1 callersFunctionfreeze_params

(params)

src/diffusers_training.py:287

↓ 1 callersFunctionfreeze_params

(params)

src/diffusers_training_sdxl.py:341

↓ 1 callersFunctiongdupdateWexact

(K, V, Ktarget1, Vtarget1, W, device='cuda')

src/composenW.py:40

↓ 1 callersFunctiongdupdateWexact

(K, V, Ktarget1, Vtarget1, W, device='cuda')

src/diffusers_composenW.py:16

↓ 1 callersFunctionget_clip_score

get standard image-text clipscore. images can either be: - a list of strings specifying filepaths for images - a precomputed, ordered

customconcept101/evaluate.py:123

↓ 1 callersFunctionget_full_repo_name

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

src/diffusers_training.py:584

↓ 1 callersMethodget_input_withmask

(self, batch, **args)

src/model.py:265

↓ 1 callersFunctionget_model

(path)

src/composenW.py:34

↓ 1 callersFunctionget_parser

(**parser_kwargs)

train.py:117

↓ 1 callersFunctiongetlayers

(model, root_name='')

src/composenW.py:76

↓ 1 callersFunctionimport_model_class_from_model_name_or_path

(pretrained_model_name_or_path: str, revision: str)

src/diffusers_training.py:292

↓ 1 callersFunctionload_model_from_config

(config, ckpt, verbose=False)

sample.py:111

↓ 1 callersFunctionload_model_from_config

(config, ckpt)

src/composenW.py:23

↓ 1 callersFunctionload_model_from_config

(config, ckpt, verbose=False)

src/convert.py:14

↓ 1 callersFunctionload_model_from_config_addtoken

(config, ckpt, verbose=False)

src/convert.py:27

↓ 1 callersMethodlog_images

(self, batch, N=8, n_row=4, sample=True, ddim_steps=200, ddim_eta=1., return_keys=None, qua

src/model.py:304

↓ 1 callersMethodlog_img

(self, pl_module, batch, batch_idx, split="train")

train.py:544

↓ 1 callersMethodlog_local

(self, save_dir, split, images, global_step, current_epoch, batch_idx)

train.py:525

↓ 1 callersFunctionmain

()

sample.py:135

↓ 1 callersFunctionmain

(path, newtoken=0)

src/get_deltas.py:9

↓ 1 callersFunctionmain

(args)

src/diffusers_training.py:594

↓ 1 callersFunctionmain

(args)

src/diffusers_training_sdxl.py:739

↓ 1 callersFunctionmain

(args)

customconcept101/evaluate.py:290

↓ 1 callersFunctionmelk

(*args, **kwargs)

train.py:946

↓ 1 callersFunctionnondefault_trainer_args

(opt)

train.py:308

↓ 1 callersFunctionparse_args

()

src/compress.py:57

↓ 1 callersFunctionparse_args

()

src/retrieve.py:79

↓ 1 callersFunctionparse_args

()

src/diffusers_sample.py:58

↓ 1 callersFunctionparse_args

()

src/composenW.py:268

↓ 1 callersFunctionparse_args

()

src/get_deltas.py:41

↓ 1 callersFunctionparse_args

()

src/diffusers_composenW.py:167

↓ 1 callersFunctionparse_args

(input_args=None)

src/diffusers_training.py:312

↓ 1 callersFunctionparse_args

(input_args=None)

src/diffusers_training_sdxl.py:402

↓ 1 callersFunctionparse_args

()

src/convert.py:141

↓ 1 callersFunctionparse_args

()

customconcept101/evaluate.py:277

↓ 1 callersMethodprepare_data

(self)

train.py:380

↓ 1 callersFunctionretrieve

(target_name, outpath, num_class_images)

src/retrieve.py:14

↓ 1 callersFunctionsample

(ckpt, delta_ckpt, from_file, prompt, compress, batch_size, freeze_model, sdxl=False)

src/diffusers_sample.py:18

↓ 1 callersFunctionsave_model_card

( repo_id: str, images=None, base_model=str, train_text_encoder=False, prompt=str, repo_folder=None, vae_p

src/diffusers_training_sdxl.py:346

↓ 1 callersMethodsetup

customconcept101/evaluate.py:37

Method__call__

( self, attn: Attention, hidden_states, encoder_hidden_states=None, at

src/diffusers_model_pipeline.py:280

Method__call__

(self, attn: Attention, hidden_states, encoder_hidden_states=None, attention_mask=None)

src/diffusers_model_pipeline.py:327

src/diffusers_data_pipeline.py:268

Method__getitem__

(self, index)

src/diffusers_data_pipeline.py:336

Method__getitem__

(self, i)

src/finetune_data.py:182

Method__getitem__

(self, idx)

customconcept101/evaluate.py:28

Method__getitem__

(self, idx)

customconcept101/evaluate.py:57

Method__getitem__

(self, idx)

customconcept101/evaluate.py:82

(self, batch_size, train=None, train2=None, validation=None, test=None, predict=None, wrap=Fa

train.py:356

Method__init__

(self, resume, now, logdir, ckptdir, cfgdir, config, lightning_config)

train.py:444

Method__init__

(self, prompt, num_samples)

src/diffusers_data_pipeline.py:261

Method__init__

( self, concepts_list, tokenizer, size=512, center_crop=False,

src/diffusers_data_pipeline.py:281

Method__init__

(self, datapath, reg_datapath=None, caption=None,

src/finetune_data.py:116

Method__init__

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

src/diffusers_model_pipeline.py:324

Method__init__

( self, vae: AutoencoderKL, text_encoder: CLIPTextModel, text_encoder_2: CLIPT

src/diffusers_model_pipeline.py:545

Method__init__

(self, modifier_token, version="openai/clip-vit-large-patch14", device="cuda", max_length=77)

src/custom_modules.py:228

Method__init__

(self, freeze_model='crossattn-kv', cond_stage_trainable=False,

src/model.py:107

Method__init__

(self, data, append=False, prefix='A photo depicts')

customconcept101/evaluate.py:20

next →1–100 of 135, ranked by callers