MCPcopy Create free account

hub / github.com/brycedrennan/imaginAIry / functions

Functions2,992 in github.com/brycedrennan/imaginAIry

↓ 352 callersMethodappend
(self, module: Module)
imaginairy/vendored/refiners/fluxion/layers/chain.py:416
↓ 331 callersMethodto
(self: T, device: Device | str | None = None, dtype: DType | None = None)
imaginairy/vendored/refiners/fluxion/layers/module.py:44
↓ 110 callersMethodget
(self, key)
imaginairy/utils/model_cache.py:279
↓ 107 callersFunctionget_device
Return the best torch backend available.
imaginairy/utils/__init__.py:21
↓ 78 callersMethodkeys
(self)
imaginairy/utils/model_cache.py:317
↓ 57 callersMethodload_state_dict
Loads the class's state. Args: state_dict (dict): scaler state. Should be an object returned from a call to :meth:
imaginairy/vendored/k_diffusion/utils.py:142
↓ 49 callersMethodeval
(self)
imaginairy/enhancers/video_interpolation/rife/RIFE_HDv3.py:11
↓ 48 callersMethodpop
(self, key)
imaginairy/utils/model_cache.py:63
↓ 46 callersFunctiondefault
(val, d)
imaginairy/utils/__init__.py:283
↓ 45 callersMethodreplace
Replace a module in the chain with a new module.
imaginairy/vendored/refiners/fluxion/layers/chain.py:442
↓ 43 callersFunctioninstantiate_from_config
Instantiate an object from a config dict.
imaginairy/utils/__init__.py:68
↓ 37 callersMethodload
Load model from file. Args: path (str): file path
imaginairy/modules/midas/midas/base_model.py:9
↓ 34 callersFunctionto_torch
(x)
imaginairy/samplers/base.py:127
↓ 34 callersMethodupdate
This method must be called once for each frame even with empty detections. NOTE:as in practical realtime MOT, the detector doesn't run
imaginairy/vendored/facexlib/tracking/sort.py:20
↓ 31 callersMethodfrom_pretrained
(cls, name="vgg_lpips")
imaginairy/modules/sgm/autoencoding/lpips/loss/lpips.py:36
↓ 31 callersMethodload
Load the LoRA weights from a dictionary of tensors. Expects the keys to be in the commonly found formats on CivitAI's hub.
imaginairy/vendored/refiners/foundationals/latent_diffusion/lora.py:29
↓ 31 callersMethodpad
(input: str, /, padding_length: int = 2)
imaginairy/vendored/refiners/foundationals/latent_diffusion/lora.py:101
↓ 29 callersMethodsigma
(self, t)
imaginairy/vendored/k_diffusion/sampling.py:488
↓ 28 callersMethodconvert
Args: box: can be a k-tuple, k-list or an Nxk array/tensor, where k = 4 or 5 from_mode, to_mode (BoxMode) Re
imaginairy/img_processors/densepose.py:370
↓ 28 callersFunctionget_cached_url_path
Gets the contents of a url, but caches the response indefinitely. While we attempt to use the cached_path from huggingface transformers, we
imaginairy/utils/downloads.py:29
↓ 27 callersMethodensure_find
(self, layer_type: type[T])
imaginairy/vendored/refiners/fluxion/layers/chain.py:375
↓ 27 callersFunctionextract_into_tensor
(a, t, x_shape)
imaginairy/modules/diffusion/util.py:132
↓ 27 callersFunctionlog
(text)
imaginairy/cli/unslow_the_cli.py:12
↓ 27 callersMethodnorm
(c_in)
imaginairy/vendored/k_diffusion/models/image_v1.py:60
↓ 27 callersMethodsort
(a, b)
imaginairy/vendored/k_diffusion/sampling.py:92
↓ 27 callersMethodtiming
(self, description)
imaginairy/utils/log_utils.py:237
↓ 26 callersFunction_n2p
(w, t=True)
imaginairy/vendored/blip/vit.py:286
↓ 25 callersMethodpop
(self, index: int = -1)
imaginairy/vendored/refiners/fluxion/layers/chain.py:419
↓ 25 callersMethodt
(self, sigma)
imaginairy/vendored/k_diffusion/sampling.py:485
↓ 23 callersFunctionconv_nd
Create a 1D, 2D, or 3D convolution module.
imaginairy/modules/diffusion/util.py:254
↓ 23 callersMethodsave
(self, save_path: "Path | str", image_type: str = "generated")
imaginairy/schema.py:929
↓ 22 callersFunctionimagine
Generates images based on the provided prompts using the ImaginAIry API. Args: prompts (list[ImaginePrompt] | str | ImaginePrompt):
imaginairy/api/generate.py:202
↓ 22 callersFunctionlog_img
(img, description)
imaginairy/utils/log_utils.py:39
↓ 22 callersFunctionlog_latent
(latents, description)
imaginairy/utils/log_utils.py:29
↓ 22 callersFunctionsigma_fn
(t)
imaginairy/vendored/k_diffusion/sampling.py:804
↓ 22 callersMethoduse_context
Retrieve the context object from the module's context provider.
imaginairy/vendored/refiners/fluxion/layers/module.py:133
↓ 20 callersMethodinsert
(self, index: int, module: Module)
imaginairy/vendored/refiners/fluxion/layers/chain.py:392
↓ 20 callersMethodnormalize
(self, coordinates: Tensor, target_size: tuple[int, int], original_size: tuple[int, int])
imaginairy/vendored/refiners/foundationals/segment_anything/model.py:135
↓ 19 callersMethodpaste
(self, tensor: Tensor, /, crop: Tensor)
imaginairy/vendored/refiners/foundationals/latent_diffusion/multi_diffusion.py:31
↓ 19 callersMethodset_context
(self, context: str, value: Any)
imaginairy/vendored/refiners/fluxion/layers/chain.py:184
↓ 19 callersMethodto
(self: T, device: Device | str | None = None, dtype: DType | None = None)
imaginairy/vendored/refiners/foundationals/latent_diffusion/schedulers/scheduler.py:130
↓ 18 callersMethod_log
Print a message if `verbose` is `True`.
imaginairy/vendored/refiners/fluxion/model_converter.py:391
↓ 18 callersFunctionappend_dims
Appends dimensions to the end of a tensor until it has target_dims dimensions.
imaginairy/vendored/k_diffusion/utils.py:40
↓ 18 callersFunctionassert_image_similar_to_expectation
(img, img_path, threshold=100)
tests/utils.py:7
↓ 18 callersFunctionpillow_fit_image_within
( image: PIL.Image.Image | LazyLoadingImage, max_height=512, max_width=512, convert="RGB",
imaginairy/utils/img_utils.py:27
↓ 17 callersFunctionload_weight_map
(map_name: str)
imaginairy/weight_management/translators.py:13
↓ 17 callersMethodtokenizer
(self)
imaginairy/vendored/refiners/foundationals/latent_diffusion/stable_diffusion_xl/text_encoder.py:47
↓ 17 callersMethodwrite
(self, *args)
imaginairy/vendored/k_diffusion/utils.py:378
↓ 16 callersMethod__init__
(self, normalize_input=False, **kwargs)
imaginairy/modules/sgm/encoders/modules.py:211
↓ 16 callersMethodsetup_adapter
(self, target: T)
imaginairy/vendored/refiners/fluxion/adapters/adapter.py:23
↓ 15 callersMethod__init__
(self, *shape: int)
imaginairy/vendored/refiners/fluxion/layers/basics.py:17
↓ 15 callersMethodfree_memory
(cls, amount)
tests/test_utils/test_memory_tracker.py:16
↓ 15 callersFunctionlinear
Create a linear module.
imaginairy/modules/sgm/diffusionmodules/util.py:286
↓ 15 callersFunctionpillow_img_to_torch_image
( img: PIL.Image.Image | LazyLoadingImage, convert="RGB" )
imaginairy/utils/img_utils.py:54
↓ 15 callersMethodremove
Remove a module from the chain.
imaginairy/vendored/refiners/fluxion/layers/chain.py:431
↓ 14 callersMethod__init__
( self, *, ch, out_ch, ch_mult=(1, 2, 4, 8), num_res_blocks,
imaginairy/modules/diffusion/model.py:376
↓ 14 callersMethodallocate_memory
(cls, amount)
tests/test_utils/test_memory_tracker.py:11
↓ 14 callersFunctionconv_nd
Create a 1D, 2D, or 3D convolution module.
imaginairy/modules/sgm/diffusionmodules/util.py:272
↓ 14 callersMethodlayers
(self, layer_type: type[T], recurse: bool = False)
imaginairy/vendored/refiners/fluxion/layers/chain.py:368
↓ 14 callersFunctiontorch_img_to_pillow_img
(img_t: torch.Tensor)
imaginairy/utils/img_utils.py:103
↓ 13 callersMethod__init__
(self, config, layer_num)
imaginairy/vendored/blip/med.py:361
↓ 13 callersFunctionconv_dw
(inp, oup, stride, leaky=0.1)
imaginairy/vendored/facexlib/detection/retinaface_net.py:25
↓ 13 callersMethodset
(self, key, model, memory_usage=0)
imaginairy/utils/model_cache.py:300
↓ 12 callersMethod__init__
(self, config, layer_num)
imaginairy/vendored/blip/nlvr_encoder.py:407
↓ 12 callersMethoddecode_first_stage
(self, z, predict_cids=False)
imaginairy/modules/diffusion/ddpm.py:1121
↓ 12 callersMethodformat
(self, record)
imaginairy/utils/log_utils.py:377
↓ 12 callersFunctionget_block
(in_channel, depth, num_units, stride=2)
imaginairy/vendored/facexlib/recognition/arcface_arch.py:81
↓ 12 callersFunctionto_d
Converts a denoiser output to a Karras ODE derivative.
imaginairy/vendored/k_diffusion/sampling.py:49
↓ 11 callersMethod__init__
( self, embedding_dim: int = 768, patch_size: int = 16, image_size: int = 224,
imaginairy/vendored/refiners/foundationals/dinov2/vit.py:233
↓ 11 callersMethodstate_dict
Returns the state of the class as a :class:`dict`.
imaginairy/vendored/k_diffusion/utils.py:138
↓ 11 callersMethodunet
(self)
imaginairy/vendored/refiners/foundationals/latent_diffusion/lora.py:18
↓ 10 callersMethod__init__
(self, dim_in, dim_out)
imaginairy/modules/sgm/attention.py:90
↓ 10 callersMethod__init__
(self, d_blocks, u_blocks, skip_stages=0)
imaginairy/vendored/k_diffusion/layers.py:283
↓ 10 callersMethod__init__
(self, in_channels: int = 768, device: Device | str | None = None, dtype: DType | None = None)
imaginairy/vendored/refiners/foundationals/segment_anything/image_encoder.py:287
↓ 10 callersFunctioncast_weights
( source_weights, source_model_name: str, source_component_name: str, source_format: str,
imaginairy/weight_management/conversion.py:109
↓ 10 callersFunctioncreate_model_of_n_bytes
(n)
tests/test_utils/test_model_cache.py:19
↓ 10 callersMethoddecode
(self, z)
imaginairy/modules/autoencoder.py:199
↓ 10 callersMethoddevice
(self)
imaginairy/utils/model_cache.py:138
↓ 10 callersFunctionlog_conditioning
(conditioning, description)
imaginairy/utils/log_utils.py:22
↓ 10 callersMethodsplit_to_multi_head
( self, x: Float[Tensor, "batch_size sequence_length embedding_dim"] )
imaginairy/vendored/refiners/fluxion/layers/attentions.py:104
↓ 9 callersFunctionNormalize
(in_channels, num_groups=32)
imaginairy/modules/diffusion/model.py:51
↓ 9 callersMethod__init__
( self, in_channels, nf, emb_dim, ch_mult, num_res_blocks,
imaginairy/vendored/codeformer/vqgan_arch.py:251
↓ 9 callersMethodcompose
(self, func: Callable[[Any], Any])
imaginairy/vendored/refiners/fluxion/layers/chain.py:68
↓ 9 callersFunctionconfigure_logging
(level="INFO")
imaginairy/utils/log_utils.py:398
↓ 9 callersFunctioncreate_vit
( vit, image_size, use_grad_checkpointing=False, ckpt_layer=0, drop_path_rate=0 )
imaginairy/vendored/blip/blip.py:240
↓ 9 callersMethodsave
(self, path)
imaginairy/weight_management/translation.py:71
↓ 9 callersFunctionseed_everything
(seed: int | None = None)
imaginairy/utils/__init__.py:342
↓ 9 callersFunctiont_fn
(sigma)
imaginairy/vendored/k_diffusion/sampling.py:807
↓ 9 callersMethodtransform
(self, image, use_origin_size)
imaginairy/vendored/facexlib/detection/retinaface.py:165
↓ 8 callersMethod__init__
(self, channels, use_conv, dims=2, out_channels=None, padding=1)
imaginairy/modules/diffusion/openaimodel.py:104
↓ 8 callersMethod__init__
( self, channels: int, use_conv: bool, dims: int = 2, out_channels: Op
imaginairy/modules/sgm/diffusionmodules/openaimodel.py:126
↓ 8 callersMethod__init__
( self, *, ch, out_ch, ch_mult=(1, 2, 4, 8), num_res_blocks,
imaginairy/modules/sgm/diffusionmodules/model.py:321
↓ 8 callersMethod__init__
(self, channels, reduction)
imaginairy/vendored/facexlib/recognition/arcface_arch.py:23
↓ 8 callersMethod__init__
( self, target: T, clip_image_encoder: CLIPImageEncoderH, image_proj: fl.Modul
imaginairy/vendored/refiners/foundationals/latent_diffusion/image_prompt.py:337
↓ 8 callersMethod__init__
(self, in_channels: int, device: Device | str | None = None, dtype: DType | None = None)
imaginairy/vendored/refiners/foundationals/latent_diffusion/stable_diffusion_1/unet.py:237
↓ 8 callersMethod__init__
( self, embedding_dim: int = 256, feed_forward_dim: int = 2048, num_layers: in
imaginairy/vendored/refiners/foundationals/segment_anything/mask_decoder.py:191
↓ 8 callersFunction_make_scratch
(in_shape, out_shape, groups=1, expand=False)
imaginairy/modules/midas/midas/blocks.py:126
↓ 8 callersMethod_set_parent
(self, parent: "Chain | None")
imaginairy/vendored/refiners/fluxion/layers/module.py:116
↓ 8 callersMethodadd_command
(self, cmd, name)
imaginairy/cli/clickshell_mod.py:59
↓ 8 callersMethodas_base64
(self)
imaginairy/schema.py:207
next →1–100 of 2,992, ranked by callers