MCPcopy Create free account

hub / github.com/IrisRainbowNeko/DreamArtist-sd-webui-extension / functions

Functions61 in github.com/IrisRainbowNeko/DreamArtist-sd-webui-extension

↓ 7 callersFunctionget_convs
(block)
scripts/dream_artist/cptuning.py:457
↓ 5 callersMethodsave
(self, filename)
scripts/dream_artist/cptuning.py:41
↓ 4 callersMethodcreate_text
(self, filename_text, idx=None)
scripts/dream_artist/dataset.py:149
↓ 2 callersMethod__init__
(self, dim, drop_path=0., layer_scale_init_value=1e-6)
scripts/dream_artist/convnext.py:26
↓ 2 callersMethodcreate_text_nofw
(self, idx=None)
scripts/dream_artist/dataset.py:155
↓ 2 callersFunctionsave_embedding
(embedding, checkpoint, embedding_name, filename, remove_cached_checksum=True, use_negative=False, embedding_n
scripts/dream_artist/cptuning.py:712
↓ 2 callersFunctionset_seed
(seed)
scripts/dream_artist/cptuning.py:342
↓ 2 callersMethodshuffle
(self)
scripts/dream_artist/dataset.py:146
↓ 2 callersMethodupdate_model_average
(self, ma_model, current_model)
scripts/dream_artist/cptuning.py:171
↓ 1 callersMethodforward_features
(self, x)
scripts/dream_artist/convnext.py:108
↓ 1 callersFunctionget_cfg_range
(cfg_text:str)
scripts/dream_artist/cptuning.py:331
↓ 1 callersMethodget_score
(self, img)
scripts/dream_artist/convnext_discriminator.py:105
↓ 1 callersFunctionload
(module, prefix='')
scripts/dream_artist/convnext_discriminator.py:46
↓ 1 callersFunctionload_state_dict
(model, state_dict, prefix='', ignore_missing="relative_position_index")
scripts/dream_artist/convnext_discriminator.py:36
↓ 1 callersMethodregister_embedding
(self, embedding, model)
scripts/dream_artist/cptuning.py:74
↓ 1 callersMethodupdate_average
(self, old, new)
scripts/dream_artist/cptuning.py:176
↓ 1 callersFunctionvalidate_train_inputs
(model_name, learn_rate, batch_size, data_root, template_file, steps, save_model_every, create_image_every, lo
scripts/dream_artist/cptuning.py:236
↓ 1 callersFunctionwrite_loss
(log_directory, filename, step, epoch_len, values)
scripts/dream_artist/cptuning.py:212
Method__call__
(self, image)
scripts/dream_artist/dataset.py:34
Method__getitem__
(self, i)
scripts/dream_artist/dataset.py:163
Method__getitem__
(self, i)
scripts/dream_artist/dataset.py:245
Method__init__
(self, in_chans=3, num_classes=1000, depths=[3, 3, 9, 3], dims=[96, 192, 384, 768], drop_pat
scripts/dream_artist/convnext.py:66
Method__init__
(self, normalized_shape, eps=1e-6, data_format="channels_last")
scripts/dream_artist/convnext.py:125
Method__init__
(self, filename=None, latent=None, filename_text=None, timg=None, att_mask=None)
scripts/dream_artist/dataset.py:18
Method__init__
(self, width, height)
scripts/dream_artist/dataset.py:30
Method__init__
(self, data_root, width, height, repeats, flip_p=0.5, placeholder_token="*", model=None, device=None, template
scripts/dream_artist/dataset.py:40
Method__init__
(self, data_root, width, height, device=None)
scripts/dream_artist/dataset.py:188
Method__init__
(self, ckpt)
scripts/dream_artist/convnext_discriminator.py:85
Method__init__
(self, vec, name, step=None)
scripts/dream_artist/cptuning.py:33
Method__init__
(self, embeddings_dir)
scripts/dream_artist/cptuning.py:68
Method__init__
(self, beta)
scripts/dream_artist/cptuning.py:167
Method__len__
(self)
scripts/dream_artist/dataset.py:160
Method__len__
(self)
scripts/dream_artist/dataset.py:242
Method_init_weights
(self, m)
scripts/dream_artist/convnext.py:103
Functionatt_hook
(module, x, output)
scripts/dream_artist/cptuning.py:759
Methodchecksum
(self)
scripts/dream_artist/cptuning.py:53
Methodconst_hash
(a)
scripts/dream_artist/cptuning.py:57
Functionconvnext_base
(pretrained=False, in_22k=False, **kwargs)
scripts/dream_artist/convnext.py:177
Functionconvnext_large
(pretrained=False, in_22k=False, **kwargs)
scripts/dream_artist/convnext.py:186
Functionconvnext_small
(pretrained=False,in_22k=False, **kwargs)
scripts/dream_artist/convnext.py:168
Functionconvnext_tiny
(pretrained=False,in_22k=False, **kwargs)
scripts/dream_artist/convnext.py:159
Functionconvnext_tiny
(pretrained=False,in_22k=False, pretrained_cfg=None, **kwargs)
scripts/dream_artist/convnext_discriminator.py:19
Functionconvnext_xlarge
(pretrained=False, in_22k=False, **kwargs)
scripts/dream_artist/convnext.py:195
Functioncreate_embedding
(name, initialization_text, initialization_text_neg, nvpt, overwrite_old, nvpt_neg, seed)
scripts/dream_artist/ui.py:10
Functioncreate_embedding
(name, num_vectors_per_token, overwrite_old, init_text='*')
scripts/dream_artist/cptuning.py:183
Methodfind_embedding_at_position
(self, tokens, offset)
scripts/dream_artist/cptuning.py:153
Methodforward
(self, x)
scripts/dream_artist/convnext.py:37
Methodforward
(self, x)
scripts/dream_artist/convnext.py:114
Methodforward
(self, x)
scripts/dream_artist/convnext.py:135
Methodget_all
(self, img)
scripts/dream_artist/convnext_discriminator.py:112
Methodload_words_embeddings
(self)
scripts/dream_artist/cptuning.py:88
Functionlr_lambda_cos
(current_step)
scripts/dream_artist/cptuning.py:480
Functionon_ui_tabs
()
scripts/dream_artist_main.py:22
Functionon_ui_train_tabs
(params)
scripts/dream_artist_main.py:16
Functionp_losses_hook
(x_start, cond, t, noise=None, scale=(1.0,1.0), att_mask=None, dy_cfg_f='ln')
scripts/dream_artist/cptuning.py:262
Functionproc_att
(*args)
scripts/dream_artist/ui.py:46
Functionproc_att
(id_task, data_root, training_width, training_height)
scripts/dream_artist/cptuning.py:747
Methodprocess_file
(path, filename)
scripts/dream_artist/cptuning.py:97
Functionstart_training_dreamartist
()
javascript/dream_artist.js:7
Functiontrain_embedding
(*args)
scripts/dream_artist/ui.py:23
Functiontrain_embedding
(id_task, embedding_name, seed, learn_rate, batch_size, data_root, log_directory, training_width, training_hei
scripts/dream_artist/cptuning.py:349