Functions — github.com/brycedrennan/imaginAIry

hub / github.com/brycedrennan/imaginAIry / functions

Functions2,992 in github.com/brycedrennan/imaginAIry

⨍ Functions 2,992◇ Types & classes 731↳ Endpoints 37

↓ 1 callersFunctiontile_untile

(img, tile_size, overlap_percent)

tests/test_feather_tile.py:89

↓ 1 callersMethodtimed_info

(self, msg, *args, hide_below_ms=0, **kwargs)

imaginairy/utils/log_utils.py:105

↓ 1 callersFunctionto_grayscale

(img: "Tensor")

imaginairy/img_processors/control_modes.py:237

↓ 1 callersMethodto_imagine_prompt

Converts this StableStudioInput to an ImaginePrompt.

imaginairy/http_app/stablestudio/models.py:73

↓ 1 callersFunctiontorch_image_to_depth_map

(image_t: torch.Tensor, model_type="dpt_hybrid")

imaginairy/modules/midas/api.py:317

↓ 1 callersFunctiontorch_image_to_openvcv_img

(img: torch.Tensor)

imaginairy/utils/img_utils.py:90

↓ 1 callersFunctiontorch_normal_map_to_pillow_img

(norm_map_t)

imaginairy/vendored/imaginairy_normal_map/model.py:68

↓ 1 callersFunctiontorch_version_check

()

imaginairy/utils/torch_installer.py:11

↓ 1 callersFunctiontrace_compvis_execution_order

(device=None)

imaginairy/weight_management/execution_trace.py:62

↓ 1 callersFunctiontrace_sd15_diffusers_execution_order

(device=None)

imaginairy/weight_management/execution_trace.py:91

↓ 1 callersFunctiontrain_mode

A context manager that places a model into training mode and restores the previous mode on exit.

imaginairy/vendored/k_diffusion/utils.py:70

↓ 1 callersFunctiontransfer

(model, model_weights)

imaginairy/img_processors/openpose.py:41

↓ 1 callersFunctiontransfer_audio

(sourceVideo, targetVideo)

imaginairy/enhancers/video_interpolation/rife/interpolate.py:26

↓ 1 callersFunctiontransformers_text_encoder_to_refiners_translator

()

imaginairy/weight_management/translators.py:18

↓ 1 callersFunctiontranslate_weights

( source_weights: TensorDict, weight_map: WeightTranslationMap, nan_check=False )

imaginairy/weight_management/translation.py:98

↓ 1 callersMethodtranslate_weights

( self, text_encoder_l_weights: TensorDict, text_encoder_g_weights: TensorDict )

imaginairy/weight_management/translators.py:100

↓ 1 callersFunctionunslowify_scripts

()

imaginairy/cli/unslow_the_cli.py:59

↓ 1 callersFunctionunslowify_scripts_safe

()

imaginairy/cli/unslow_the_cli.py:76

↓ 1 callersMethodup

(self)

imaginairy/vendored/refiners/fluxion/adapters/lora.py:42

↓ 1 callersFunctionupscale_image

(img, ultrasharp=False)

imaginairy/enhancers/upscale_realesrgan.py:43

↓ 1 callersMethodvalidate

(cls, v, info)

imaginairy/http_app/utils.py:32

↓ 1 callersFunctionversion

Print the version.

imaginairy/cli/main.py:56

↓ 1 callersMethodvisual_forward_masked

(self, img_s, seg_s)

imaginairy/vendored/clipseg/__init__.py:611

↓ 1 callersFunctionvisualize_detection

Visualize detection results. Args: img (Numpy array): Input image. CHW, BGR, [0, 255], uint8.

imaginairy/vendored/facexlib/visualization/vis_detection.py:5

↓ 1 callersMethodvisualize_iuv_arr

(self, image_bgr, iuv_arr: np.ndarray, bbox_xywh)

imaginairy/img_processors/densepose.py:547

↓ 1 callersFunctionwarp_and_crop_face

Function: ---------- apply affine transform 'trans' to uv Parameters: ---------- @src_img: 3x3 np.array i

imaginairy/vendored/facexlib/detection/align_trans.py:145

↓ 1 callersMethodweight_update

(self, num_tokens)

imaginairy/modules/sgm/autoencoding/regularizers/quantize.py:347

↓ 1 callersFunctionwhitespace_clean

(text)

imaginairy/vendored/clip/simple_tokenizer.py:64

↓ 1 callersMethodwith_pos_embed

(self, tensor, pos: Optional[Tensor])

imaginairy/vendored/codeformer/codeformer_arch.py:125

↓ 1 callersFunctionwrite_pfm

Write pfm file. Args: path (str): pathto file image (array): data scale (int, optional): Scale. Defaults to 1.

imaginairy/modules/midas/utils.py:75

Function_TileModeConv2DConvForward

( self, input: torch.Tensor, # noqa weight: torch.Tensor, bias: torch.Tensor, )

imaginairy/modules/diffusion/ddpm.py:716

Method__add__

(self, other: "ContextProvider")

imaginairy/vendored/refiners/fluxion/context.py:32

Method__add__

(self, other: "Chain | Module | list[Module]")

imaginairy/vendored/refiners/fluxion/layers/chain.py:288

Method__bool__

(self)

imaginairy/utils/model_cache.py:87

Method__bool__

(self)

imaginairy/vendored/refiners/fluxion/context.py:40

Method__bool__

(self)

imaginairy/vendored/refiners/fluxion/model_converter.py:123

Method__call__

(self, img)

imaginairy/modules/midas/utils.py:25

Method__call__

(self, sample)

imaginairy/modules/midas/midas/transforms.py:166

Method__call__

(self, sample)

imaginairy/modules/midas/midas/transforms.py:208

Method__call__

(self, sample)

imaginairy/modules/midas/midas/transforms.py:220

Method__call__

(self, sigma: torch.Tensor)

imaginairy/modules/sgm/diffusionmodules/loss_weighting.py:10

Method__call__

(self, sigma: torch.Tensor)

imaginairy/modules/sgm/diffusionmodules/loss_weighting.py:15

Method__call__

(self, sigma: torch.Tensor)

imaginairy/modules/sgm/diffusionmodules/loss_weighting.py:23

Method__call__

(self, sigma: torch.Tensor)

imaginairy/modules/sgm/diffusionmodules/loss_weighting.py:33

Method__call__

(self, sigma)

imaginairy/modules/sgm/diffusionmodules/denoiser_weighting.py:7

Method__call__

(self, sigma)

imaginairy/modules/sgm/diffusionmodules/denoiser_weighting.py:15

Method__call__

(self, sigma)

imaginairy/modules/sgm/diffusionmodules/denoiser_weighting.py:25

Method__call__

( self, sigma: torch.Tensor )

imaginairy/modules/sgm/diffusionmodules/denoiser_scaling.py:11

Method__call__

( self, sigma: torch.Tensor )

imaginairy/modules/sgm/diffusionmodules/denoiser_scaling.py:21

Method__call__

( self, sigma: torch.Tensor )

imaginairy/modules/sgm/diffusionmodules/denoiser_scaling.py:32

Method__call__

( self, sigma: torch.Tensor )

imaginairy/modules/sgm/diffusionmodules/denoiser_scaling.py:43

Method__call__

( self, sigma: torch.Tensor )

imaginairy/modules/sgm/diffusionmodules/denoiser_scaling.py:54

Method__call__

(self, n, do_append_zero=True, device="cpu", flip=False)

imaginairy/modules/sgm/diffusionmodules/discretizer.py:24

Method__call__

(self, denoiser, x, cond, uc=None, num_steps=None)

imaginairy/modules/sgm/diffusionmodules/sampling.py:114

Method__call__

(self, denoiser, x, cond, uc=None, num_steps=None)

imaginairy/modules/sgm/diffusionmodules/sampling.py:160

Method__call__

(self, denoiser, x, cond, uc=None, num_steps=None, **kwargs)

imaginairy/modules/sgm/diffusionmodules/sampling.py:189

Method__call__

(self, denoiser, x, cond, uc=None, num_steps=None, **kwargs)

imaginairy/modules/sgm/diffusionmodules/sampling.py:349

Method__call__

(self, n_samples, rand=None)

imaginairy/modules/sgm/diffusionmodules/sigma_sampling.py:13

Method__call__

(self, n_samples, rand=None)

imaginairy/modules/sgm/diffusionmodules/sigma_sampling.py:28

Method__call__

(self, x: torch.Tensor, sigma: float)

imaginairy/modules/sgm/diffusionmodules/guiders.py:18

Method__call__

(self, x: torch.Tensor, sigma: torch.Tensor)

imaginairy/modules/sgm/diffusionmodules/guiders.py:31

Method__call__

(self, x: torch.Tensor, sigma: float)

imaginairy/modules/sgm/diffusionmodules/guiders.py:49

Method__call__

(self, x: torch.Tensor, sigma: torch.Tensor)

imaginairy/modules/sgm/diffusionmodules/guiders.py:81

Method__call__

(self, text)

imaginairy/enhancers/upscale_riverwing.py:149

Method__call__

(self, *args, **kwargs)

imaginairy/utils/model_cache.py:366

Method__call__

(self, image)

imaginairy/vendored/k_diffusion/augmentation.py:37

Method__call__

(self, t0, t1)

imaginairy/vendored/k_diffusion/sampling.py:95

Method__call__

(self, sigma, sigma_next)

imaginairy/vendored/k_diffusion/sampling.py:123

Method__call__

(self, *args: Any)

imaginairy/vendored/refiners/fluxion/layers/chain.py:58

Method__call__

(self, x: Tensor)

imaginairy/vendored/refiners/fluxion/layers/chain.py:86

Method__call__

(self, x: Tensor, /, noise: Tensor, step: int, targets: list[D])

imaginairy/vendored/refiners/foundationals/latent_diffusion/multi_diffusion.py:46

Method__call__

( self, x: torch.Tensor, /, clip_text_embedding: torch.Tensor, conditi

imaginairy/vendored/refiners/foundationals/latent_diffusion/restart.py:48

Method__call__

(self, text: str)

imaginairy/vendored/refiners/foundationals/latent_diffusion/stable_diffusion_xl/text_encoder.py:43

Method__call__

Represents one step of the backward diffusion process that iteratively denoises the input data `x`. This method works by estimating

imaginairy/vendored/refiners/foundationals/latent_diffusion/schedulers/dpm_solver.py:97

Method__call__

(self, x: Tensor, noise: Tensor, step: int, generator: Generator | None = None)

imaginairy/vendored/refiners/foundationals/latent_diffusion/schedulers/ddim.py:39

Method__call__

Applies a step of the diffusion process to the input tensor `x` using the provided `noise` and `timestep`. This method should be ove

imaginairy/vendored/refiners/foundationals/latent_diffusion/schedulers/scheduler.py:55

Method__call__

(self, x: Tensor, noise: Tensor, step: int, generator: Generator | None = None)

imaginairy/vendored/refiners/foundationals/latent_diffusion/schedulers/ddpm.py:36

Method__call__

( self, x: Tensor, noise: Tensor, step: int, generator: Generator | No

imaginairy/vendored/refiners/foundationals/latent_diffusion/schedulers/euler.py:58

Method__contains__

(self, item)

imaginairy/utils/model_cache.py:72

Method__contains__

(self, key)

imaginairy/utils/model_cache.py:314

Method__delitem__

(self, key)

imaginairy/utils/model_cache.py:75

imaginairy/utils/memory_tracker.py:38

Method__enter__

(self)

imaginairy/utils/log_utils.py:182

Method__enter__

(self)

imaginairy/utils/log_utils.py:221

imaginairy/utils/memory_tracker.py:42

Method__exit__

(self, exc_type, exc_value, traceback)

imaginairy/utils/log_utils.py:186

Method__exit__

(self, exc_type, exc_val, exc_tb)

imaginairy/utils/log_utils.py:224

Method__get_pydantic_core_schema__

( cls, source_type: Any, handler: GetCoreSchemaHandler )

imaginairy/schema.py:146

Method__get_validators__

(cls)

imaginairy/http_app/utils.py:28

Method__getattr__

(self, key)

imaginairy/schema.py:106

Method__getattr__

(self, name)

imaginairy/utils/model_cache.py:362

Method__getattr__

(self, name: str)

imaginairy/vendored/refiners/fluxion/layers/module.py:30

Method__getitem__

(self, item)

imaginairy/utils/model_cache.py:78

Method__getitem__

(self, key)

imaginairy/utils/model_cache.py:297

Method__getitem__

(self, key)

imaginairy/vendored/k_diffusion/utils.py:360

Method__getitem__

(self, key: int)

imaginairy/vendored/refiners/fluxion/layers/chain.py:296

Function__init__

( self, target: SD1UNet, name: str, weights_location: str, )

imaginairy/modules/refiners_sd.py:843

Method__init__

(self, name)

tests/utils.py:31

← previous next →1,201–1,300 of 2,992, ranked by callers