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, in_features)
tests/test_utils/test_model_cache.py:11
Method__init__
( self, *, filepath: str | None = None, url: str | None = None, img: "
imaginairy/schema.py:67
Method__init__
( self, prompt: PromptInput = "", *, negative_prompt: PromptInput = None,
imaginairy/schema.py:375
Method__init__
( self, img, prompt: ImaginePrompt, is_nsfw, safety_score, res
imaginairy/schema.py:841
Method__init__
( self, version="google/t5-v1_1-large", device="cuda", max_length=77, freeze=True )
imaginairy/modules/encoders.py:61
Method__init__
( self, version="openai/clip-vit-large-patch14", device="cuda", max_length=77,
imaginairy/modules/encoders.py:103
Method__init__
( self, arch="ViT-H-14", version="laion2b_s32b_b79k", device=None, max
imaginairy/modules/encoders.py:169
Method__init__
( self, clip_version="openai/clip-vit-large-patch14", t5_version="google/t5-v1_1-xl",
imaginairy/modules/encoders.py:237
Method__init__
(self, model, decay=0.9999, use_num_upates=True)
imaginairy/modules/ema.py:10
Method__init__
( self, unet: SDXLUNet | None = None, lda: SDXLAutoencoder | None = None, clip
imaginairy/modules/refiners_sd.py:321
Method__init__
( self, unet: SDXLUNet | None = None, lda: SDXLAutoencoder | None = None, clip
imaginairy/modules/refiners_sd.py:547
Method__init__
(self, dim, dim_out=None, mult=4, glu=False, dropout=0.0)
imaginairy/modules/attention.py:42
Method__init__
(self, dim, heads=4, dim_head=32)
imaginairy/modules/attention.py:76
Method__init__
(self, query_dim, context_dim=None, heads=8, dim_head=64, dropout=0.0)
imaginairy/modules/attention.py:121
Method__init__
(self, query_dim, context_dim=None, heads=8, dim_head=64, dropout=0.0)
imaginairy/modules/attention.py:258
Method__init__
( self, dim, n_heads, d_head, dropout=0.0, context_dim=None,
imaginairy/modules/attention.py:317
Method__init__
( self, in_channels, n_heads, d_head, depth=1, dropout=0.0,
imaginairy/modules/attention.py:381
Method__init__
( self, control_stage_config, control_key, only_mid_control, *args,
imaginairy/modules/cldm.py:366
Method__init__
(self, *args, vq_interface=False, **kwargs)
imaginairy/modules/autoencoder.py:591
Method__init__
(self, parameters, deterministic=False)
imaginairy/modules/distributions.py:8
Method__init__
( self, version="ViT-L/14", device=get_device(), max_length=77, n_repe
imaginairy/modules/clip_embedders.py:59
Method__init__
( self, model_name, jit=False, device=get_device(), antialias=False,
imaginairy/modules/clip_embedders.py:99
Method__init__
( self, spacial_dim: int, embed_dim: int, num_heads_channels: int, out
imaginairy/modules/diffusion/openaimodel.py:39
Method__init__
(self, channels, out_channels=None, ks=5)
imaginairy/modules/diffusion/openaimodel.py:131
Method__init__
(self, channels, use_conv, dims=2, out_channels=None, padding=1)
imaginairy/modules/diffusion/openaimodel.py:153
Method__init__
( self, channels, emb_channels, dropout, out_channels=None, us
imaginairy/modules/diffusion/openaimodel.py:194
Method__init__
( self, channels, num_heads=1, num_head_channels=-1, use_checkpoint=Fa
imaginairy/modules/diffusion/openaimodel.py:299
Method__init__
(self, n_heads)
imaginairy/modules/diffusion/openaimodel.py:368
Method__init__
(self, n_heads)
imaginairy/modules/diffusion/openaimodel.py:400
Method__init__
( self, image_size, in_channels, model_channels, out_channels,
imaginairy/modules/diffusion/openaimodel.py:459
Method__init__
(self, noise_schedule_config=None)
imaginairy/modules/diffusion/upscaling.py:14
Method__init__
(self, noise_schedule_config, max_noise_level=1000, to_cuda=False)
imaginairy/modules/diffusion/upscaling.py:95
Method__init__
( self, first_stage_config, cond_stage_config, num_timesteps_cond=None,
imaginairy/modules/diffusion/ddpm.py:738
Method__init__
(self, diff_model_config, conditioning_key)
imaginairy/modules/diffusion/ddpm.py:1715
Method__init__
( self, concat_keys: tuple, finetune_keys=( "model.diffusion_model.input_b
imaginairy/modules/diffusion/ddpm.py:1751
Method__init__
( self, concat_keys=("mask", "masked_image"), masked_image_key="masked_image",
imaginairy/modules/diffusion/ddpm.py:1933
Method__init__
( self, concat_keys=("lr",), reshuffle_patch_size=None, low_scale_config=None,
imaginairy/modules/diffusion/ddpm.py:1992
Method__init__
(self, c_concat_config, c_crossattn_config)
imaginairy/modules/diffusion/util.py:288
Method__init__
(self, in_channels, with_conv)
imaginairy/modules/diffusion/model.py:58
Method__init__
(self, in_channels, with_conv)
imaginairy/modules/diffusion/model.py:74
Method__init__
( self, *, in_channels, out_channels=None, conv_shortcut=False,
imaginairy/modules/diffusion/model.py:94
Method__init__
(self, in_channels)
imaginairy/modules/diffusion/model.py:166
Method__init__
(self, in_channels)
imaginairy/modules/diffusion/model.py:287
Method__init__
( self, *, ch, out_ch, ch_mult=(1, 2, 4, 8), num_res_blocks,
imaginairy/modules/diffusion/model.py:551
Method__init__
( self, *, ch, out_ch, ch_mult=(1, 2, 4, 8), num_res_blocks,
imaginairy/modules/diffusion/model.py:668
Method__init__
(self, in_channels, out_channels, *args, **kwargs)
imaginairy/modules/diffusion/model.py:822
Method__init__
( self, in_channels, out_channels, ch, num_res_blocks, resolut
imaginairy/modules/diffusion/model.py:866
Method__init__
(self, factor, in_channels, mid_channels, out_channels, depth=2)
imaginairy/modules/diffusion/model.py:924
Method__init__
( self, in_channels, ch, resolution, out_ch, num_res_blocks,
imaginairy/modules/diffusion/model.py:980
Method__init__
( self, z_channels, out_ch, resolution, num_res_blocks, attn_r
imaginairy/modules/diffusion/model.py:1024
Method__init__
(self, in_size, out_size, in_channels, out_channels, ch_mult=2)
imaginairy/modules/diffusion/model.py:1067
Method__init__
(self, in_channels=None, learned=False, mode="bilinear")
imaginairy/modules/diffusion/model.py:1099
Method__init__
(self, model_type)
imaginairy/modules/midas/api.py:335
Method__init__
(self, model_type="dpt_hybrid")
imaginairy/modules/midas/utils.py:14
Method__init__
Init. Args: path (str, optional): Path to saved model. Defaults to None. features (int, optional): Number of
imaginairy/modules/midas/midas/midas_net_custom.py:17
Method__init__
Init. Args: features (int): number of features
imaginairy/modules/midas/midas/blocks.py:264
Method__init__
Init. Args: features (int): number of features
imaginairy/modules/midas/midas/blocks.py:302
Method__init__
Init. Args: features (int): number of features
imaginairy/modules/midas/midas/blocks.py:336
Method__init__
Init. Args: features (int): number of features
imaginairy/modules/midas/midas/blocks.py:407
Method__init__
Init. Args: width (int): desired output width height (int): desired output height resize_target
imaginairy/modules/midas/midas/transforms.py:55
Method__init__
(self, mean, std)
imaginairy/modules/midas/midas/transforms.py:204
Method__init__
(self)
imaginairy/modules/midas/midas/transforms.py:217
Method__init__
Init. Args: path (str, optional): Path to saved model. Defaults to None. features (int, optional): Number of
imaginairy/modules/midas/midas/midas_net.py:17
Method__init__
(self, path=None, non_negative=True, **kwargs)
imaginairy/modules/midas/midas/dpt_depth.py:147
Method__init__
(self, start_index=1)
imaginairy/modules/midas/midas/backbones/utils.py:15
Method__init__
(self, in_features, start_index=1)
imaginairy/modules/midas/midas/backbones/utils.py:28
Method__init__
(self, dim0, dim1)
imaginairy/modules/midas/midas/backbones/utils.py:42
Method__init__
( self, in_channels, n_heads, d_head, depth=1, dropout=0.0,
imaginairy/modules/sgm/video_attention.py:178
Method__init__
( self, network_config, denoiser_config, first_stage_config, condition
imaginairy/modules/sgm/diffusion.py:34
Method__init__
(self, dim, dim_out=None, mult=4, glu=False, dropout=0.0)
imaginairy/modules/sgm/attention.py:100
Method__init__
(self, dim, heads=4, dim_head=32)
imaginairy/modules/sgm/attention.py:134
Method__init__
( self, dim: int, num_heads: int = 8, qkv_bias: bool = False, qk_scale
imaginairy/modules/sgm/attention.py:159
Method__init__
( self, query_dim, context_dim=None, heads=8, dim_head=64, dro
imaginairy/modules/sgm/attention.py:213
Method__init__
( self, query_dim, context_dim=None, heads=8, dim_head=64, dro
imaginairy/modules/sgm/attention.py:288
Method__init__
( self, query_dim, context_dim=None, heads=8, dim_head=64, dropout=0.0, **kwargs )
imaginairy/modules/sgm/attention.py:394
Method__init__
( self, dim, n_heads, d_head, dropout=0.0, context_dim=None,
imaginairy/modules/sgm/attention.py:513
Method__init__
( self, dim, n_heads, d_head, dropout=0.0, context_dim=None,
imaginairy/modules/sgm/attention.py:632
Method__init__
( self, in_channels, n_heads, d_head, depth=1, dropout=0.0,
imaginairy/modules/sgm/attention.py:679
Method__init__
( self, dim: int, depth: int, heads: int, dim_head: int, conte
imaginairy/modules/sgm/attention.py:777
Method__init__
( self, ema_decay: Union[None, float] = None, monitor: Union[None, str] = None,
imaginairy/modules/sgm/autoencoder.py:41
Method__init__
( self, *args, encoder_config: Dict, decoder_config: Dict, loss_config
imaginairy/modules/sgm/autoencoder.py:119
Method__init__
(self, embed_dim: int, **kwargs)
imaginairy/modules/sgm/autoencoder.py:451
Method__init__
( self, embed_dim: int, n_embed: int, sane_index_shape: bool = False,
imaginairy/modules/sgm/autoencoder.py:537
Method__init__
(self, *args, **kwargs)
imaginairy/modules/sgm/autoencoder.py:564
Method__init__
( self, model: nn.Module, shape: Union[None, Tuple[int, int], List[int]] = (16, 16),
imaginairy/modules/sgm/autoencoder.py:578
Method__init__
(self, **kwargs)
imaginairy/modules/sgm/autoencoder.py:616
Method__init__
( self, sigma_sampler_config: dict, loss_weighting_config: dict, loss_type: st
imaginairy/modules/sgm/diffusionmodules/loss.py:17
Method__init__
(self, sigma_data: float = 0.5)
imaginairy/modules/sgm/diffusionmodules/loss_weighting.py:20
Method__init__
(self)
imaginairy/modules/sgm/diffusionmodules/loss_weighting.py:28
Method__init__
(self, sigma_data=0.5)
imaginairy/modules/sgm/diffusionmodules/denoiser_weighting.py:12
Method__init__
(self)
imaginairy/modules/sgm/diffusionmodules/denoiser_weighting.py:20
Method__init__
( self, spacial_dim: int, embed_dim: int, num_heads_channels: int, out
imaginairy/modules/sgm/diffusionmodules/openaimodel.py:37
Method__init__
( self, channels: int, use_conv: bool, dims: int = 2, out_channels: Op
imaginairy/modules/sgm/diffusionmodules/openaimodel.py:179
Method__init__
( self, channels: int, emb_channels: int, dropout: float, out_channels
imaginairy/modules/sgm/diffusionmodules/openaimodel.py:236
Method__init__
( self, channels: int, num_heads: int = 1, num_head_channels: int = -1,
imaginairy/modules/sgm/diffusionmodules/openaimodel.py:374
Method__init__
(self, n_heads: int)
imaginairy/modules/sgm/diffusionmodules/openaimodel.py:420
Method__init__
(self, n_heads: int)
imaginairy/modules/sgm/diffusionmodules/openaimodel.py:448
Method__init__
(self, dim: int)
imaginairy/modules/sgm/diffusionmodules/openaimodel.py:474
Method__init__
( self, in_channels: int, model_channels: int, out_channels: int, num_
imaginairy/modules/sgm/diffusionmodules/openaimodel.py:512
Method__init__
(self, sigma_data: float = 0.5)
imaginairy/modules/sgm/diffusionmodules/denoiser_scaling.py:18
← previousnext →1,301–1,400 of 2,992, ranked by callers