Functions — github.com/Monalissaa/DisenDiff

hub / github.com/Monalissaa/DisenDiff / functions

Functions238 in github.com/Monalissaa/DisenDiff

⨍ Functions 238◇ Types & classes 39

↓ 6 callersMethod__init__

(self, embed_dim: int, # vision image_resolution: int,

clip/model.py:250

↓ 5 callersMethodencode

(self, *args, **kwargs)

src/custom_modules.py:221

↓ 4 callersMethod_make_layer

(self, planes, blocks, stride=1)

clip/model.py:126

↓ 4 callersMethoddecode

(self, tokens)

clip/simple_tokenizer.py:129

↓ 4 callersFunctionisimage

(path)

src/finetune_data.py:110

↓ 4 callersMethodtokenize

(self, strings: list)

utils/clip_eval.py:23

↓ 3 callersMethodencode_image

(self, image)

clip/model.py:346

↓ 3 callersFunctionextract_all_captions

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

utils/clipscore/clipscore.py:107

↓ 3 callersMethodget_empty_store

()

src/attention_controllers.py:10

↓ 3 callersMethodget_empty_store

()

src/attention_controllers.py:179

↓ 3 callersMethodget_image_features

(self, img: torch.Tensor, norm: bool = True)

utils/clip_eval.py:46

↓ 3 callersFunctionmulti_head_attention_forward

(query: Tensor, key: Tensor, value: Tensor,

clip/auxilary.py:26

↓ 3 callersFunctionpreprocess

(image, scale, resample)

src/diffusers_data_pipeline.py:226

↓ 3 callersFunctionregister_recr

(net_, count, place_in_unet)

src/ptp_utils.py:222

↓ 3 callersMethodsave_pretrained

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

src/diffusers_model_pipeline.py:448

↓ 3 callersMethodshared_step

(self, batch, **kwargs)

src/model.py:583

↓ 2 callersMethod__init__

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

train.py:516

↓ 2 callersFunction_transform

(n_px)

clip/clip.py:58

↓ 2 callersMethodbetween_steps

(self)

src/attention_controllers.py:189

↓ 2 callersFunctionbytes_to_unicode

Returns list of utf-8 byte and a corresponding list of unicode strings. The reversible bpe codes work on unicode strings. This means you

clip/simple_tokenizer.py:16

↓ 2 callersFunctioncompute_human_correlation

(input_json, image_directory, tauvariant='c')

utils/clipscore/flickr8k_example/compute_metrics.py:17

↓ 2 callersFunctiondiffusion_step

(model, controller, latents, context, t, guidance_scale, low_resource=False)

src/ptp_utils.py:64

↓ 2 callersMethodencode_text

(self, tokens: list)

utils/clip_eval.py:27

↓ 2 callersFunctionextract_all_images

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

utils/clipscore/clipscore.py:120

↓ 2 callersMethodforward

(self, attn, is_cross: bool, place_in_unet: str)

src/attention_controllers.py:146

↓ 2 callersMethodget_average_attention

(self)

src/attention_controllers.py:210

↓ 2 callersFunctionget_pairs

Return set of symbol pairs in a word. Word is represented as tuple of symbols (symbols being variable-length strings).

clip/simple_tokenizer.py:38

↓ 2 callersFunctionget_text_embedding

(prompts)

src/composenW.py:92

↓ 2 callersFunctionget_text_embedding

(prompts)

src/diffusers_composenW.py:55

↓ 2 callersFunctioninit_latent

(latent, model, height, width, generator, batch_size)

src/ptp_utils.py:87

↓ 2 callersFunctionlatent2image

(vae, latents)

src/ptp_utils.py:78

↓ 2 callersMethodload_model

(self, save_path, compress=False)

src/diffusers_model_pipeline.py:472

↓ 2 callersFunctionpatch_device

(module)

clip/clip.py:123

↓ 2 callersFunctionpatch_float

(module)

clip/clip.py:143

↓ 2 callersFunctionregister_attention_control

(model, controller)

src/ptp_utils.py:174

↓ 2 callersFunctiontext_under_image

(image: np.ndarray, text: str, text_color: Tuple[int, int, int] = (0, 0, 0))

sample.py:146

↓ 2 callersFunctionupdate_alpha_time_word

(alpha, bounds: Union[float, Tuple[float, float]], prompt_ind: int, word_inds: Opti

src/ptp_utils.py:265

↓ 2 callersFunctionview_images

(images, save_path, base_count, num_rows=1, offset_ratio=0.02, step=None)

sample.py:158

↓ 1 callersMethod__init__

(self, device, clip_model='ViT-B/32')

utils/clip_eval.py:13

↓ 1 callersMethod__init__

(self, in_features: int, out_features: int)

clip/auxilary.py:23

↓ 1 callersMethod__init__

(self, channels, kernel_size, dim=2)

src/gaussian_smoothing.py:87

↓ 1 callersMethod__init__

(self)

src/custom_modules.py:218

↓ 1 callersFunction_download

(url: str, root: str = os.path.expanduser("~/.cache/clip"))

clip/clip.py:26

↓ 1 callersMethod_reset_parameters

(self)

clip/auxilary.py:335

↓ 1 callersMethod_transform_test

(self, n_px)

utils/clipscore/clipscore.py:88

↓ 1 callersMethodadd_token

(self)

src/custom_modules.py:242

↓ 1 callersFunctionaggregate_attention_all

(prompts, attention_store: AttentionStore, res: int, from_where: List[str], is_cross: bool, select: int, len_t

sample.py:202

↓ 1 callersFunctionaggregate_attention_all_step

(prompts, attention_store: AttentionStore, res: int, from_where: List[str], is_cross: bool, select: int, len_t

sample.py:230

↓ 1 callersMethodattention

(self, x: torch.Tensor)

clip/model.py:190

↓ 1 callersFunctionavailable_models

Returns the names of available CLIP models

clip/clip.py:68

↓ 1 callersFunctionbasic_clean

(text)

clip/simple_tokenizer.py:50

↓ 1 callersMethodbetween_steps

(self)

src/attention_controllers.py:138

↓ 1 callersMethodbpe

(self, token)

clip/simple_tokenizer.py:80

↓ 1 callersMethodbuild_attention_mask

(self)

clip/model.py:334

↓ 1 callersFunctionbuild_model

(state_dict: dict)

clip/model.py:405

↓ 1 callersFunctionca_forward

(self, place_in_unet)

src/ptp_utils.py:175

↓ 1 callersFunctionchange_attn

(unet)

src/diffusers_training.py:274

↓ 1 callersMethodcheck_frequency

(self, check_idx)

train.py:599

↓ 1 callersFunctioncollate_fn

(examples, with_prior_preservation)

src/diffusers_data_pipeline.py:233

↓ 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 callersFunctionconvert_weights

Convert applicable model parameters to fp16

clip/model.py:381

↓ 1 callersFunctioncreate_custom_diffusion

(unet, freeze_model)

src/diffusers_training.py:254

↓ 1 callersMethodcustom_forward

r""" Returns:

src/custom_modules.py:257

↓ 1 callersFunctiondefault_bpe

()

clip/simple_tokenizer.py:11

↓ 1 callersMethodencode

(self, text)

clip/simple_tokenizer.py:121

↓ 1 callersMethodencode_images

(self, images: torch.Tensor)

utils/clip_eval.py:31

↓ 1 callersMethodencode_text

(self, text)

clip/model.py:349

↓ 1 callersMethodevaluate

(self, gen_samples, src_images, target_text)

utils/clip_eval.py:112

↓ 1 callersMethodforward

(self, image, text)

clip/model.py:364

↓ 1 callersMethodfreeze

(self)

src/custom_modules.py:282

↓ 1 callersFunctionfreeze_params

(params)

src/diffusers_training.py:287

↓ 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

utils/clipscore/clipscore.py:135

↓ 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:548

↓ 1 callersFunctionget_model

(path)

src/composenW.py:34

↓ 1 callersFunctionget_parser

(**parser_kwargs)

train.py:117

↓ 1 callersFunctionget_refonlyclipscore

The text only side for refclipscore

utils/clipscore/clipscore.py:162

↓ 1 callersMethodget_text_features

(self, text: str, norm: bool = True)

utils/clip_eval.py:35

↓ 1 callersFunctionget_word_inds

(text: str, word_place: int, tokenizer)

src/ptp_utils.py:244

↓ 1 callersFunctiongetlayers

(model, root_name='')

src/composenW.py:76

↓ 1 callersMethodimg_to_img_similarity

(self, src_images, generated_images)

utils/clip_eval.py:54

↓ 1 callersFunctionimport_model_class_from_model_name_or_path

(pretrained_model_name_or_path: str, revision: str)

src/diffusers_training.py:292

↓ 1 callersMethodinitialize_parameters

(self)

clip/model.py:305

↓ 1 callersFunctionload_model_from_config

(config, ckpt, verbose=False)

sample.py:115

↓ 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:590

↓ 1 callersMethodlog_img

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

train.py:567

↓ 1 callersMethodlog_local

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

train.py:548

↓ 1 callersFunctionmain

()

sample.py:298

↓ 1 callersFunctionmain

()

utils/clipscore/clipscore.py:205

↓ 1 callersFunctionmain

()

utils/clipscore/flickr8k_example/download.py:140

↓ 1 callersFunctionmain

()

utils/clipscore/flickr8k_example/compute_metrics.py:74

↓ 1 callersFunctionmain

(path, newtoken=0)

src/get_deltas.py:9

next →1–100 of 238, ranked by callers