MCPcopy Create free account

hub / github.com/brycedrennan/imaginAIry / functions

Functions2,992 in github.com/brycedrennan/imaginAIry

⨍ Functions 2,992◇ Types & classes 731↳ Endpoints 37
Method__init__
( self, linear_start=0.00085, linear_end=0.0120, num_timesteps=1000, )
imaginairy/modules/sgm/diffusionmodules/discretizer.py:49
Method__init__
( self, discretization_config: Union[Dict, ListConfig, OmegaConf], num_steps: Union[in
imaginairy/modules/sgm/diffusionmodules/sampling.py:27
Method__init__
(self, eta=1.0, s_noise=1.0, *args, **kwargs)
imaginairy/modules/sgm/diffusionmodules/sampling.py:139
Method__init__
( self, order=4, *args, **kwargs, )
imaginairy/modules/sgm/diffusionmodules/sampling.py:179
Method__init__
( self, alpha: float, merge_strategy: str = "learned_with_images", rearrange_p
imaginairy/modules/sgm/diffusionmodules/util.py:310
Method__init__
(self, p_mean=-1.2, p_std=1.2)
imaginairy/modules/sgm/diffusionmodules/sigma_sampling.py:9
Method__init__
(self, discretization_config, num_idx, do_append_zero=False, flip=True)
imaginairy/modules/sgm/diffusionmodules/sigma_sampling.py:19
Method__init__
( self, channels: int, emb_channels: int, dropout: float, video_kernel
imaginairy/modules/sgm/diffusionmodules/video_model.py:32
Method__init__
( self, scaling_config: Dict, num_idx: int, discretization_config: Dict,
imaginairy/modules/sgm/diffusionmodules/denoiser.py:49
Method__init__
(self, diffusion_model, compile_model: bool = False)
imaginairy/modules/sgm/diffusionmodules/wrappers.py:11
Method__init__
(self, in_channels, with_conv)
imaginairy/modules/sgm/diffusionmodules/model.py:61
Method__init__
(self, in_channels, with_conv)
imaginairy/modules/sgm/diffusionmodules/model.py:77
Method__init__
( self, *, in_channels, out_channels=None, conv_shortcut=False,
imaginairy/modules/sgm/diffusionmodules/model.py:97
Method__init__
(self, in_channels)
imaginairy/modules/sgm/diffusionmodules/model.py:159
Method__init__
(self, in_channels)
imaginairy/modules/sgm/diffusionmodules/model.py:164
Method__init__
(self, in_channels)
imaginairy/modules/sgm/diffusionmodules/model.py:214
Method__init__
( self, *, ch, out_ch, ch_mult=(1, 2, 4, 8), num_res_blocks,
imaginairy/modules/sgm/diffusionmodules/model.py:496
Method__init__
( self, *, ch, out_ch, ch_mult=(1, 2, 4, 8), num_res_blocks,
imaginairy/modules/sgm/diffusionmodules/model.py:613
Method__init__
(self, scale: float)
imaginairy/modules/sgm/diffusionmodules/guiders.py:28
Method__init__
( self, max_scale: float, num_frames: int, min_scale: float = 1.0, add
imaginairy/modules/sgm/diffusionmodules/guiders.py:64
Method__init__
( self, out_channels, *args, dropout=0.0, video_kernel_size=3,
imaginairy/modules/sgm/autoencoding/temporal_ae.py:25
Method__init__
( self, in_channels: int, alpha: float = 0, merge_strategy: str = "learned" )
imaginairy/modules/sgm/autoencoding/temporal_ae.py:118
Method__init__
( self, in_channels: int, alpha: float = 0, merge_strategy: str = "learned" )
imaginairy/modules/sgm/autoencoding/temporal_ae.py:190
Method__init__
( self, *args, video_kernel_size: Union[int, list] = 3, alpha: float = 0.0,
imaginairy/modules/sgm/autoencoding/temporal_ae.py:305
Method__init__
(self)
imaginairy/modules/sgm/autoencoding/regularizers/base.py:12
Method__init__
(self, sample: bool = True)
imaginairy/modules/sgm/autoencoding/regularizers/__init__.py:13
Method__init__
(self)
imaginairy/modules/sgm/autoencoding/regularizers/quantize.py:20
Method__init__
( self, num_hiddens: int, embedding_dim: int, n_embed: int, straight_t
imaginairy/modules/sgm/autoencoding/regularizers/quantize.py:75
Method__init__
( self, n_e: int, e_dim: int, beta: float = 0.25, remap: Optional[str]
imaginairy/modules/sgm/autoencoding/regularizers/quantize.py:186
Method__init__
( self, n_embed: int, embedding_dim: int, beta: float, decay: float =
imaginairy/modules/sgm/autoencoding/regularizers/quantize.py:358
Method__init__
( self, input_dim: int, n_codes: int, codebook_dim: int, beta: float =
imaginairy/modules/sgm/autoencoding/regularizers/quantize.py:449
Method__init__
( self, num_features, logdet=False, affine=True, allow_reverse_init=False )
imaginairy/modules/sgm/autoencoding/lpips/util.py:47
Method__init__
Construct a PatchGAN discriminator Parameters: input_nc (int) -- the number of channels in input images ndf (int)
imaginairy/modules/sgm/autoencoding/lpips/model/model.py:24
Method__init__
(self)
imaginairy/modules/sgm/autoencoding/lpips/loss/lpips.py:72
Method__init__
(self, chn_in, chn_out=1, use_dropout=False)
imaginairy/modules/sgm/autoencoding/lpips/loss/lpips.py:88
Method__init__
(self, requires_grad=False, pretrained=True)
imaginairy/modules/sgm/autoencoding/lpips/loss/lpips.py:104
Method__init__
( self, disc_start: int, logvar_init: float = 0.0, disc_num_layers: int = 3,
imaginairy/modules/sgm/autoencoding/losses/discriminator_loss.py:21
Method__init__
( self, decoder_config, perceptual_weight=1.0, latent_weight=1.0, scal
imaginairy/modules/sgm/autoencoding/losses/lpips.py:11
Method__init__
(self, value)
imaginairy/modules/sgm/distributions/distributions.py:16
Method__init__
(self, parameters, deterministic=False)
imaginairy/modules/sgm/distributions/distributions.py:27
Method__init__
(self)
imaginairy/modules/sgm/encoders/modules.py:49
Method__init__
(self, emb_models: Union[List, ListConfig])
imaginairy/modules/sgm/encoders/modules.py:96
Method__init__
(self, embed_dim, n_classes=1000, add_sequence_dim=False)
imaginairy/modules/sgm/encoders/modules.py:236
Method__init__
( self, version="google/t5-v1_1-xxl", device=None, max_length=77, freeze=True )
imaginairy/modules/sgm/encoders/modules.py:273
Method__init__
( self, version="google/byt5-base", device=None, max_length=77, freeze=True )
imaginairy/modules/sgm/encoders/modules.py:316
Method__init__
( self, version="openai/clip-vit-large-patch14", device=None, max_length=77,
imaginairy/modules/sgm/encoders/modules.py:359
Method__init__
( self, arch="ViT-H-14", version="laion2b_s32b_b79k", device=None, max
imaginairy/modules/sgm/encoders/modules.py:427
Method__init__
( self, arch="ViT-H-14", version="laion2b_s32b_b79k", device=None, max
imaginairy/modules/sgm/encoders/modules.py:530
Method__init__
( self, arch="ViT-H-14", version="laion2b_s32b_b79k", device=None, max
imaginairy/modules/sgm/encoders/modules.py:601
Method__init__
( self, clip_version="openai/clip-vit-large-patch14", t5_version="google/t5-v1_1-xl",
imaginairy/modules/sgm/encoders/modules.py:756
Method__init__
( self, n_stages=1, method="bilinear", multiplier=0.5, in_channels=3,
imaginairy/modules/sgm/encoders/modules.py:785
Method__init__
( self, model_config, linear_start, linear_end, timesteps=1000,
imaginairy/modules/sgm/encoders/modules.py:845
Method__init__
(self, outdim)
imaginairy/modules/sgm/encoders/modules.py:941
Method__init__
( self, weight: float = 1.0, flatten_output: bool = True, *args, **kwargs )
imaginairy/modules/sgm/encoders/modules.py:958
Method__init__
( self, n_cond_frames: int, n_copies: int, encoder_config: dict, sigma
imaginairy/modules/sgm/encoders/modules.py:977
Method__init__
( self, open_clip_embedding_config: Dict, n_cond_frames: int, n_copies: int,
imaginairy/modules/sgm/encoders/modules.py:1054
Method__init__
(self, text)
imaginairy/enhancers/bool_masker.py:44
Method__init__
(self, mask, modifier)
imaginairy/enhancers/bool_masker.py:71
Method__init__
(self, masks, op)
imaginairy/enhancers/bool_masker.py:105
Method__init__
(self, inner_model, sigma_data=1.0, embed_dim=256)
imaginairy/enhancers/upscale_riverwing.py:20
Method__init__
(self, version="openai/clip-vit-large-patch14", max_length=77)
imaginairy/enhancers/upscale_riverwing.py:143
Method__init__
(self, version="openai/clip-vit-large-patch14", device="cuda")
imaginairy/enhancers/upscale_riverwing.py:168
Method__init__
(self, window_size=11, size_average=True, channel=3)
imaginairy/enhancers/video_interpolation/rife/msssim.py:289
Method__init__
(self)
imaginairy/enhancers/video_interpolation/rife/RIFE_HDv3.py:7
Method__init__
(self, c, dilation=1)
imaginairy/enhancers/video_interpolation/rife/IFNet_HDv3.py:64
Method__init__
(self, in_planes, c=64)
imaginairy/enhancers/video_interpolation/rife/IFNet_HDv3.py:75
Method__init__
(self)
imaginairy/enhancers/video_interpolation/rife/IFNet_HDv3.py:120
Method__init__
(self)
imaginairy/utils/safety.py:21
Method__init__
(self)
imaginairy/utils/spaced_kv_parser.py:65
Method__init__
( self, name="", callback_fn: "Callable[[TorchRAMTracker], None] | None" = None )
imaginairy/utils/memory_tracker.py:13
Method__init__
(self, *args, **kwargs)
imaginairy/utils/model_cache.py:35
Method__init__
(self, fn, namespace, estimated_ram_size_mb, *args, **kwargs)
imaginairy/utils/model_cache.py:334
Method__init__
( self, description: str, device: str | None = None, callback_fn: Callable | N
imaginairy/utils/log_utils.py:133
Method__init__
( self, prompt, model=None, debug_img_callback=None, img_outdir=None,
imaginairy/utils/log_utils.py:191
Method__init__
(self, data, add_data_values=True)
imaginairy/utils/data_distorter.py:80
Method__init__
(self, model)
imaginairy/samplers/kdiff.py:66
Method__init__
(self, model)
imaginairy/samplers/base.py:43
Method__init__
( self, model_num_timesteps, model_alphas_cumprod, ddim_num_steps, ddi
imaginairy/samplers/base.py:132
Method__init__
(self, model)
imaginairy/samplers/editing.py:15
Method__init__
(self, *args, **kwargs)
imaginairy/cli/clickshell_mod.py:87
Method__init__
(self)
imaginairy/weight_management/translators.py:71
Method__init__
(self)
imaginairy/img_processors/hed_boundary.py:14
Method__init__
(self)
imaginairy/img_processors/openpose.py:420
Method__init__
( self, inplace=True, cmap=cv2.COLORMAP_PARULA, val_scale=1.0, alpha=0
imaginairy/img_processors/densepose.py:457
Method__init__
( self, data_extractor, segm_extractor, inplace=True, cmap=cv2.COLORMA
imaginairy/img_processors/densepose.py:519
Method__init__
( self, detail_weight=0.2, edge_radius=0.4, edge_weight=-20, outside_i
imaginairy/vendored/smart_crop.py:39
Method__init__
( self, scale, model_path, dni_weight=None, model=None, tile=0
imaginairy/vendored/realesrgan.py:33
Method__init__
(self, img_list, num_prefetch_queue)
imaginairy/vendored/realesrgan.py:317
Method__init__
(self, inner_model, sigma_data=1.0)
imaginairy/vendored/k_diffusion/layers.py:15
Method__init__
(self, *main, skip=None)
imaginairy/vendored/k_diffusion/layers.py:61
Method__init__
(self, module)
imaginairy/vendored/k_diffusion/layers.py:78
Method__init__
(self, *main, skip=None)
imaginairy/vendored/k_diffusion/layers.py:97
Method__init__
(self, feats_in, c_out, num_groups, eps=1e-5, cond_key="cond")
imaginairy/vendored/k_diffusion/layers.py:112
Method__init__
(self, c_in, n_head, norm, dropout_rate=0.0)
imaginairy/vendored/k_diffusion/layers.py:133
Method__init__
( self, c_dec, c_enc, n_head, norm_dec, dropout_rate=0.0,
imaginairy/vendored/k_diffusion/layers.py:155
Method__init__
(self, kernel="linear", pad_mode="reflect")
imaginairy/vendored/k_diffusion/layers.py:228
Method__init__
(self, kernel="linear", pad_mode="reflect")
imaginairy/vendored/k_diffusion/layers.py:246
Method__init__
(self, in_features, out_features, std=1.0)
imaginairy/vendored/k_diffusion/layers.py:267
Method__init__
(self, a_prob=0.12, a_scale=2**0.2, a_aniso=2**0.2, a_trans=1 / 8)
imaginairy/vendored/k_diffusion/augmentation.py:31
Method__init__
(self, model)
imaginairy/vendored/k_diffusion/augmentation.py:96
← previousnext →1,401–1,500 of 2,992, ranked by callers