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
Methodforward
(self, x: Tensor)
imaginairy/vendored/refiners/fluxion/layers/basics.py:168
Methodforward
(self, x: Tensor)
imaginairy/vendored/refiners/fluxion/layers/basics.py:182
Methodforward
(self, _: Tensor)
imaginairy/vendored/refiners/fluxion/layers/basics.py:206
Methodforward
(self, x: Int[Tensor, "batch length"])
imaginairy/vendored/refiners/fluxion/layers/embedding.py:20
Methodforward
(self, x: Tensor)
imaginairy/vendored/refiners/foundationals/latent_diffusion/freeu.py:49
Methodforward
( self, key_value: Float[Tensor, "batch sequence_length 2*head_dim*num_heads"], query:
imaginairy/vendored/refiners/foundationals/latent_diffusion/image_prompt.py:88
Methodforward
( self, x: Tensor, step: int, *, clip_text_embedding: Tensor, condition_scale: float = 7.5, **kwargs:
imaginairy/vendored/refiners/foundationals/latent_diffusion/model.py:90
Methodforward
( self, x: Tensor, step: int, *, clip_text_embedding: Tensor,
imaginairy/vendored/refiners/foundationals/latent_diffusion/stable_diffusion_xl/model.py:79
Methodforward
( self, x: Tensor, step: int, *, clip_text_embedding: Tensor, condition_scale: float = 7.5, **_: Tenso
imaginairy/vendored/refiners/foundationals/latent_diffusion/stable_diffusion_1/model.py:124
Methodforward
(self, iou_mask_tokens: Tensor)
imaginairy/vendored/refiners/foundationals/segment_anything/mask_decoder.py:17
Methodforward
(self)
imaginairy/vendored/refiners/foundationals/segment_anything/mask_decoder.py:132
Methodforward
(self, x: Tensor)
imaginairy/vendored/refiners/foundationals/segment_anything/image_encoder.py:87
Methodforward
(self, x: Tensor)
imaginairy/vendored/refiners/foundationals/segment_anything/image_encoder.py:206
Methodforward
(self, x: Tensor)
imaginairy/vendored/refiners/foundationals/segment_anything/image_encoder.py:226
Methodforward
(self, type_mask: Int[Tensor, "1 num_points"])
imaginairy/vendored/refiners/foundationals/segment_anything/prompt_encoder.py:45
Methodforward
(self, text: str)
imaginairy/vendored/refiners/foundationals/clip/tokenizer.py:54
Methodforward
(self, x: Tensor)
imaginairy/vendored/refiners/foundationals/dinov2/vit.py:75
Methodforward
(self, x, mask=None)
imaginairy/vendored/codeformer/codeformer_arch.py:68
Methodforward
( self, tgt, tgt_mask: Optional[Tensor] = None, tgt_key_padding_mask: Optional
imaginairy/vendored/codeformer/codeformer_arch.py:128
Methodforward
(self, enc_feat, dec_feat, w=1)
imaginairy/vendored/codeformer/codeformer_arch.py:167
Methodforward
(self, x, w=0, detach_16=True, code_only=False, adain=False)
imaginairy/vendored/codeformer/codeformer_arch.py:261
Methodforward
(self, z)
imaginairy/vendored/codeformer/vqgan_arch.py:35
Methodforward
(self, z)
imaginairy/vendored/codeformer/vqgan_arch.py:124
Methodforward
(self, x)
imaginairy/vendored/codeformer/vqgan_arch.py:151
Methodforward
(self, x)
imaginairy/vendored/codeformer/vqgan_arch.py:165
Methodforward
(self, x_in)
imaginairy/vendored/codeformer/vqgan_arch.py:190
Methodforward
(self, x)
imaginairy/vendored/codeformer/vqgan_arch.py:223
Methodforward
(self, x)
imaginairy/vendored/codeformer/vqgan_arch.py:301
Methodforward
(self, x)
imaginairy/vendored/codeformer/vqgan_arch.py:356
Methodforward
(self, x)
imaginairy/vendored/codeformer/vqgan_arch.py:439
Methodforward
(self, x)
imaginairy/vendored/codeformer/vqgan_arch.py:507
Functionforward_beit
(pretrained, x)
imaginairy/modules/midas/midas/backbones/beit.py:14
Functionforward_default
(pretrained, x, function_name="forward_features")
imaginairy/modules/midas/midas/backbones/utils.py:62
Functionforward_flex
(self, x)
imaginairy/modules/midas/midas/backbones/vit.py:40
Functionforward_vit
(pretrained, x)
imaginairy/modules/midas/midas/backbones/vit.py:19
Methodfreeze
(self)
imaginairy/modules/clip_embedders.py:74
Methodfreeze_norm
(self)
imaginairy/vendored/facexlib/matting/modnet.py:251
Methodfrom_and
(cls, instring, tokens_start, ret_tokens)
imaginairy/enhancers/bool_masker.py:115
Methodfrom_modifier_parse
(cls, instring, tokens_start, ret_tokens)
imaginairy/enhancers/bool_masker.py:81
Methodfrom_not
(cls, instring, tokens_start, ret_tokens)
imaginairy/enhancers/bool_masker.py:120
Methodfrom_or
(cls, instring, tokens_start, ret_tokens)
imaginairy/enhancers/bool_masker.py:110
Functionfrom_pil_image
Converts from a PIL image to a tensor.
imaginairy/vendored/k_diffusion/utils.py:16
Methodfrom_simple_prompt
(cls, instring, tokens_start, ret_tokens)
imaginairy/enhancers/bool_masker.py:48
Methodfrom_weights
( cls, down: Tensor, up: Tensor, )
imaginairy/vendored/refiners/fluxion/adapters/lora.py:113
Methodfrom_weights
( cls, down: Tensor, up: Tensor, )
imaginairy/vendored/refiners/fluxion/adapters/lora.py:182
Functionfuse_model
(m)
imaginairy/modules/midas/midas/midas_net_custom.py:158
Methodgather_text_descriptions
(self)
imaginairy/enhancers/bool_masker.py:54
Methodgather_text_descriptions
(self)
imaginairy/enhancers/bool_masker.py:87
Methodgather_text_descriptions
(self)
imaginairy/enhancers/bool_masker.py:131
Functiongenerate
(studio_request: StableStudioBatchRequest)
imaginairy/http_app/stablestudio/routes.py:21
Functiongenerate_image
(imagine_prompt)
scripts/modal_entrypoint.py:63
Functiongenerate_image_b64
ImaginePrompt to generated base64 encoded image.
imaginairy/http_app/utils.py:19
Methodgenerate_offset_grid
(size: tuple[int, int], stride: int = 8)
imaginairy/vendored/refiners/foundationals/latent_diffusion/multi_diffusion.py:90
Functiongenerate_redir
()
imaginairy/http_app/app.py:64
Functiongenerate_single_image
( prompt: ImaginePrompt, debug_img_callback=None, progress_img_callback=None, progress_img_int
imaginairy/api/generate_flux.py:90
Methodget
(self, item)
imaginairy/utils/model_cache.py:81
Methodget_attention_layer
( ch, num_heads, dim_head, depth=1, context_dim=No
imaginairy/modules/sgm/diffusionmodules/video_model.py:217
Methodget_attention_map
(self)
imaginairy/vendored/blip/vit.py:79
Methodget_attention_map
(self)
imaginairy/vendored/blip/nlvr_encoder.py:132
Methodget_attention_map
(self)
imaginairy/vendored/blip/med.py:142
Methodget_attn_gradients
(self)
imaginairy/vendored/blip/vit.py:73
Methodget_attn_gradients
(self)
imaginairy/vendored/blip/nlvr_encoder.py:126
Methodget_attn_gradients
(self)
imaginairy/vendored/blip/med.py:136
Methodget_autoencoder_params
(self)
imaginairy/modules/sgm/autoencoder.py:477
Methodget_codebook_entry
( self, indices: torch.Tensor, shape: Optional[Tuple[int, ...]] = None )
imaginairy/modules/sgm/autoencoding/regularizers/quantize.py:57
Methodget_codebook_entry
( self, indices: torch.Tensor, shape: Optional[Tuple[int, ...]] = None )
imaginairy/modules/sgm/autoencoding/regularizers/quantize.py:304
Functionget_default_dtype
Return the default dtype for torch.
imaginairy/utils/__init__.py:33
Methodget_eps
(self, *args, **kwargs)
imaginairy/vendored/k_diffusion/external.py:146
Methodget_eps
(self, *args, **kwargs)
imaginairy/vendored/k_diffusion/external.py:159
Methodget_gns
Returns the current gradient noise scale.
imaginairy/vendored/k_diffusion/gns.py:105
Methodget_input
( self, batch, k, cond_key=None, bs=None, return_first_stage_outputs=False )
imaginairy/modules/diffusion/ddpm.py:1947
Methodget_input
( self, batch, k, cond_key=None, bs=None, return_first_stage_outputs=False )
imaginairy/modules/diffusion/ddpm.py:2017
Methodget_input
(self, batch)
imaginairy/modules/sgm/autoencoder.py:73
Methodget_input
(self, x: Any)
imaginairy/modules/sgm/autoencoder.py:567
Methodget_input_embeddings
(self)
imaginairy/vendored/blip/nlvr_encoder.py:694
Methodget_input_embeddings
(self)
imaginairy/vendored/blip/med.py:647
Methodget_landmarks
(self, img)
imaginairy/vendored/facexlib/alignment/awing_arch.py:359
Methodget_last_layer
(self)
imaginairy/modules/diffusion/model.py:546
Methodget_last_layer
(self)
imaginairy/modules/sgm/video_attention.py:173
Methodget_last_layer
(self)
imaginairy/modules/sgm/diffusionmodules/model.py:491
Methodget_last_layer
(self, **kwargs)
imaginairy/modules/sgm/diffusionmodules/model.py:720
Methodget_last_layer
(self, skip_time_mix=False, **kwargs)
imaginairy/modules/sgm/autoencoding/temporal_ae.py:323
Methodget_lr
(self)
imaginairy/vendored/k_diffusion/utils.py:196
Methodget_lr
(self)
imaginairy/vendored/k_diffusion/utils.py:248
Methodget_mapping
Get the mapping between the source and target models' state_dicts.
imaginairy/vendored/refiners/fluxion/model_converter.py:227
Methodget_mask_for_image
(self, img)
imaginairy/enhancers/bool_masker.py:33
Methodget_output_embeddings
(self)
imaginairy/vendored/blip/med.py:922
Methodget_position_ids
(self, x: Tensor)
imaginairy/vendored/refiners/foundationals/clip/common.py:30
Methodget_resblock
( merge_factor, merge_strategy, video_kernel_size, ch,
imaginairy/modules/sgm/diffusionmodules/video_model.py:246
Methodget_sigmas
(self, n, device="cpu")
imaginairy/modules/sgm/diffusionmodules/discretizer.py:40
Methodget_sigmas
(self, n, device="cpu")
imaginairy/modules/sgm/diffusionmodules/discretizer.py:64
Functionget_sigmas_exponential
Constructs an exponential noise schedule.
imaginairy/vendored/k_diffusion/sampling.py:25
Functionget_sigmas_karras
Constructs the noise schedule of Karras et al. (2022).
imaginairy/vendored/k_diffusion/sampling.py:16
Functionget_sigmas_polyexponential
Constructs an polynomial in log sigma noise schedule.
imaginairy/vendored/k_diffusion/sampling.py:33
Functionget_sigmas_vp
Constructs a continuous VP noise schedule.
imaginairy/vendored/k_diffusion/sampling.py:42
Methodget_stats
(self)
imaginairy/vendored/k_diffusion/gns.py:33
Methodget_stats
Returns the current (debiased) estimates of the squared mean gradient and gradient variance.
imaginairy/vendored/k_diffusion/gns.py:109
Methodget_trainable_parameters
(self)
imaginairy/modules/sgm/autoencoding/regularizers/base.py:27
Methodget_trainable_parameters
(self)
imaginairy/modules/sgm/autoencoding/regularizers/__init__.py:17
Methodget_trainable_parameters
(self)
imaginairy/modules/sgm/autoencoding/regularizers/quantize.py:62
← previousnext →2,401–2,500 of 2,992, ranked by callers