MCPcopy Create free account

hub / github.com/brycedrennan/imaginAIry / functions

Functions2,992 in github.com/brycedrennan/imaginAIry

↓ 2 callersFunction_make_vit_b16_backbone
( model, features=[96, 192, 384, 768], size=[384, 384], hooks=[2, 5, 8, 11], vit_features=
imaginairy/modules/midas/midas/backbones/vit.py:82
↓ 2 callersMethod_move_to_gpu
(self, key, model)
imaginairy/utils/model_cache.py:232
↓ 2 callersMethod_sliced_attention
(self, q, k, v, mask, slice_size)
imaginairy/modules/sgm/attention.py:372
↓ 2 callersMethod_store_nth_residual
(self, n: int)
imaginairy/vendored/refiners/foundationals/latent_diffusion/stable_diffusion_1/controlnet.py:133
↓ 2 callersFunction_totensor
(img, bgr2rgb, float32)
imaginairy/vendored/facexlib/utils/misc.py:43
↓ 2 callersFunction_totensor
(img, bgr2rgb, float32)
imaginairy/vendored/basicsr/img_util.py:26
↓ 2 callersMethod_trace_module_execution_order
Execute a forward pass and store the order of execution of specific sub-modules. - `module`: The module to trace. - `args`:
imaginairy/vendored/refiners/fluxion/model_converter.py:516
↓ 2 callersFunction_transform
(n_px)
imaginairy/vendored/clip/clip.py:94
↓ 2 callersMethod_unpack_module_args
Unpack the positional and keyword arguments passed to a module.
imaginairy/vendored/refiners/fluxion/model_converter.py:411
↓ 2 callersMethod_validation_step
(self, batch, batch_idx, postfix="")
imaginairy/modules/autoencoder.py:368
↓ 2 callersMethod_validation_step
(self, batch: dict, batch_idx: int, postfix: str = "")
imaginairy/modules/sgm/autoencoder.py:318
↓ 2 callersMethod_walk
( self, predicate: Callable[[Module, "Chain"], bool] | None = None, recurse: bool = False )
imaginairy/vendored/refiners/fluxion/layers/chain.py:333
↓ 2 callersMethodadd_nsfw_score
(self, concept_idx, abs_score, threshold)
imaginairy/utils/safety.py:39
↓ 2 callersMethodadd_restored_face
(self, face)
imaginairy/vendored/facexlib/utils/face_restoration_helper.py:278
↓ 2 callersMethodalign_warp_face
Align and warp faces with face template.
imaginairy/vendored/facexlib/utils/face_restoration_helper.py:234
↓ 2 callersMethodancestral_euler_step
(self, x, denoised, sigma, sigma_down)
imaginairy/modules/sgm/diffusionmodules/sampling.py:146
↓ 2 callersMethodancestral_step
(self, x, sigma, next_sigma, sigma_up)
imaginairy/modules/sgm/diffusionmodules/sampling.py:152
↓ 2 callersMethodapply_ckpt
(self, ckpt: Union[None, str, dict])
imaginairy/modules/sgm/autoencoder.py:61
↓ 2 callersMethodapply_model
(self, x_noisy, t, cond, *args, **kwargs)
imaginairy/modules/cldm.py:387
↓ 2 callersMethodapply_model
(self, x_noisy, t, cond, return_ids=False)
imaginairy/modules/diffusion/ddpm.py:1154
↓ 2 callersFunctionassert_tensor_float_11_bchw
(img)
imaginairy/img_processors/densepose.py:317
↓ 2 callersMethodauto_attach
( loras: dict[str, Lora], target: fl.Chain, /, exclude: list[str] | None = Non
imaginairy/vendored/refiners/foundationals/latent_diffusion/lora.py:123
↓ 2 callersFunctionbytes_to_unicode
Returns list of utf-8 byte and a corresponding list of unicode strings. The reversible bpe codes work on unicode strings. This means you
imaginairy/vendored/clip/simple_tokenizer.py:20
↓ 2 callersFunctioncalc_mean_std
Calculate mean and std for adaptive_instance_normalization. Args: feat (Tensor): 4D tensor. eps (float): A small value added to t
imaginairy/vendored/codeformer/codeformer_arch.py:13
↓ 2 callersFunctioncalc_scale_to_fit_within
(height: int, width: int, max_size)
imaginairy/utils/img_utils.py:253
↓ 2 callersFunctioncalculate_brightness
(image)
imaginairy/api/colorize.py:72
↓ 2 callersMethodcompute_clip_image_embedding
(self, image_prompt: Tensor)
imaginairy/vendored/refiners/foundationals/latent_diffusion/image_prompt.py:417
↓ 2 callersMethodcompute_conditional
(self, conditional)
imaginairy/vendored/clipseg/__init__.py:333
↓ 2 callersMethodcompute_relative_coords
(self, size: int)
imaginairy/vendored/refiners/foundationals/segment_anything/image_encoder.py:111
↓ 2 callersFunctioncompute_sinusoidal_embedding
( x: Int[Tensor, "*batch 1"], embedding_dim: int, )
imaginairy/vendored/refiners/foundationals/latent_diffusion/range_adapter.py:10
↓ 2 callersFunctionconcat_all_gather
Performs all_gather operation on the provided tensors. *** Warning ***: torch.distributed.all_gather has no gradient.
imaginairy/vendored/blip/blip_pretrain.py:320
↓ 2 callersFunctioncondition_up
(prompts)
imaginairy/enhancers/upscale_riverwing.py:201
↓ 2 callersFunctioncontrol_img_to_pillow_img
(img_t)
tests/img_processors/test_control_modes.py:17
↓ 2 callersFunctionconv
(in_planes, out_planes, kernel_size=3, stride=1, padding=1, dilation=1)
imaginairy/enhancers/video_interpolation/rife/IFNet_HDv3.py:10
↓ 2 callersFunctionconv3x3
3x3 convolution with padding
imaginairy/vendored/facexlib/parsing/resnet.py:5
↓ 2 callersFunctionconvert_bbox_to_z
Takes a bounding box in the form [x1,y1,x2,y2] and returns z in the form [x,y,s,r] where x,y is the centre of the box and s is the scale/area and
imaginairy/vendored/facexlib/tracking/kalman_tracker.py:5
↓ 2 callersFunctionconvert_x_to_bbox
Takes a bounding box in the centre form [x,y,s,r] and returns it in the form [x1,y1,x2,y2] where x1,y1 is the top left and x2,y2 is the bottom
imaginairy/vendored/facexlib/tracking/kalman_tracker.py:19
↓ 2 callersFunctioncount_flops_attn
A counter for the `thop` package to count the operations in an attention operation. Meant to be used like: macs, params = thop.pr
imaginairy/modules/diffusion/openaimodel.py:343
↓ 2 callersMethodcreate
(contexts: Contexts)
imaginairy/vendored/refiners/fluxion/context.py:27
↓ 2 callersFunctioncreate_halo_effect
( bw_image: PIL.Image.Image, background_color: tuple )
imaginairy/utils/img_utils.py:190
↓ 2 callersFunctioncreate_normal_map_torch_img
(img_t, device=get_device())
imaginairy/vendored/imaginairy_normal_map/model.py:35
↓ 2 callersFunctiondecode
(latent)
tests/test_modules/test_autoencoders.py:101
↓ 2 callersMethoddecode
(self, x: Tensor)
imaginairy/vendored/refiners/foundationals/latent_diffusion/auto_encoder.py:208
↓ 2 callersMethoddecode_all_at_once
(self, x: Tensor)
imaginairy/modules/refiners_sd.py:745
↓ 2 callersFunctiondefault_sigma
(kernel_size: int)
imaginairy/vendored/refiners/fluxion/utils.py:88
↓ 2 callersMethoddelta_border
:param h: height :param w: width :return: normalized distance to image border, wtith min distance = 0 at border and
imaginairy/modules/autoencoder.py:444
↓ 2 callersMethoddelta_border
:param h: height :param w: width :return: normalized distance to image border, wtith min distance = 0 at border and
imaginairy/modules/diffusion/ddpm.py:928
↓ 2 callersMethoddpm_solver_1_step
(self, x, t, t_next, eps_cache=None)
imaginairy/vendored/k_diffusion/sampling.py:502
↓ 2 callersMethoddpm_solver_3_step
(self, x, t, t_next, r1=1 / 3, r2=2 / 3, eps_cache=None)
imaginairy/vendored/k_diffusion/sampling.py:523
↓ 2 callersMethodema_scope
(self, context=None)
imaginairy/modules/autoencoder.py:89
↓ 2 callersMethodema_scope
(self, context=None)
imaginairy/modules/diffusion/ddpm.py:310
↓ 2 callersMethodema_scope
(self, context=None)
imaginairy/modules/sgm/autoencoder.py:82
↓ 2 callersMethodencode
(self, x: Tensor)
imaginairy/vendored/refiners/foundationals/latent_diffusion/auto_encoder.py:203
↓ 2 callersMethodencode_first_stage
(self, x)
imaginairy/modules/diffusion/ddpm.py:1133
↓ 2 callersMethodenhance
(self, img, outscale=None, alpha_upsampler="realesrgan")
imaginairy/vendored/realesrgan.py:228
↓ 2 callersMethodeuler_step
(self, x, d, dt)
imaginairy/modules/sgm/diffusionmodules/sampling.py:83
↓ 2 callersFunctionexpand_dims_like
(x, y)
imaginairy/utils/__init__.py:295
↓ 2 callersFunctionextract_into_tensor
(a, t, x_shape)
imaginairy/modules/sgm/diffusionmodules/util.py:36
↓ 2 callersFunctionface_restore_helper
Provide a singleton of FaceRestoreHelper. FaceRestoreHelper loads a model internally so we need to cache it or we end up with a memory l
imaginairy/enhancers/face_restoration_codeformer.py:84
↓ 2 callersMethodfirst_key
(self)
imaginairy/utils/model_cache.py:41
↓ 2 callersFunctionfix_torch_group_norm
Patch group_norm to cast the weights to the same type as the inputs. From what I can understand all the other repos just switch to full prec
imaginairy/utils/__init__.py:146
↓ 2 callersFunctionfix_torch_nn_layer_norm
https://github.com/CompVis/stable-diffusion/issues/25#issuecomment-1221416526.
imaginairy/utils/__init__.py:135
↓ 2 callersFunctionformat_filename
Formats the filename based on the provided template and variables.
tests/test_utils/test_format_file_name.py:6
↓ 2 callersMethodforward
(self, inp)
imaginairy/modules/sgm/encoders/modules.py:218
↓ 2 callersMethodforward
(self, *inputs: Any)
imaginairy/vendored/refiners/fluxion/layers/chain.py:525
↓ 2 callersFunctionforward_adapted_unflatten
(pretrained, x, function_name="forward_features")
imaginairy/modules/midas/midas/backbones/utils.py:82
↓ 2 callersFunctionfrange
Range but handles floats.
imaginairy/utils/__init__.py:210
↓ 2 callersFunctiongaussian
(window_size, sigma)
imaginairy/enhancers/video_interpolation/rife/msssim.py:9
↓ 2 callersMethodgenerate
( self, image, sample=False, num_beams=3, max_length=30, min_l
imaginairy/vendored/blip/blip.py:152
↓ 2 callersFunctiongenerate_video
Generates a video from a single image or multiple images, conditioned on the provided input_path. Args: input_path (str): Path to an
imaginairy/api/video_sample.py:38
↓ 2 callersFunctiongenerate_word_image
( text: str, width: int, height: int, margin_pct: float = 0.1, line_spacing: int = 4,
imaginairy/utils/text_image.py:54
↓ 2 callersFunctionget_ancestral_step
(sigma_from, sigma_to, eta=1.0)
imaginairy/modules/sgm/diffusionmodules/sampling_utils.py:25
↓ 2 callersMethodget_autoencoder_params
(self)
imaginairy/modules/sgm/autoencoder.py:188
↓ 2 callersFunctionget_cache_dir
()
imaginairy/utils/downloads.py:185
↓ 2 callersFunctionget_ckpt_path
(name, root, check=False)
imaginairy/modules/sgm/autoencoding/lpips/util.py:35
↓ 2 callersMethodget_codebook_entry
(self, indices, shape)
imaginairy/modules/sgm/autoencoding/regularizers/quantize.py:160
↓ 2 callersMethodget_cond_vec
(self, conditional, batch_size)
imaginairy/vendored/clipseg/__init__.py:302
↓ 2 callersFunctionget_debug_info
()
imaginairy/utils/debug_info.py:6
↓ 2 callersMethodget_eps
(self, *args, **kwargs)
imaginairy/vendored/k_diffusion/external.py:115
↓ 2 callersMethodget_face_landmarks_5
(self, only_keep_largest=False, only_center_face=Fal
imaginairy/vendored/facexlib/utils/face_restoration_helper.py:123
↓ 2 callersMethodget_fold_unfold
:param x: img of size (bs, c, h, w) :return: n img crops of size (n, bs, c, kernel_size[0], kernel_size[1])
imaginairy/modules/autoencoder.py:481
↓ 2 callersFunctionget_gaussian_kernel1d
(kernel_size: int, sigma: float)
imaginairy/vendored/refiners/fluxion/utils.py:73
↓ 2 callersMethodget_input
( self, batch, k, return_first_stage_outputs=False, force_c_encode=Fal
imaginairy/modules/diffusion/ddpm.py:1061
↓ 2 callersMethodget_input
(self, batch)
imaginairy/modules/sgm/diffusion.py:126
↓ 2 callersFunctionget_largest_face
(det_faces, h, w)
imaginairy/vendored/facexlib/utils/face_restoration_helper.py:12
↓ 2 callersMethodget_last_layer
(self)
imaginairy/modules/sgm/autoencoder.py:205
↓ 2 callersFunctionget_model
()
imaginairy/enhancers/describe_image_clip.py:16
↓ 2 callersFunctionget_next_filenumber
Get the next file number in a directory.
imaginairy/utils/__init__.py:249
↓ 2 callersFunctionget_pairs
Return set of symbol pairs in a word. Word is represented as tuple of symbols (symbols being variable-length strings).
imaginairy/vendored/clip/simple_tokenizer.py:46
↓ 2 callersMethodget_param_groups
( self, parameter_names: List[List[str]], optimizer_args: List[dict] )
imaginairy/modules/sgm/autoencoder.py:356
↓ 2 callersFunctionget_prompt_list
(prompt)
imaginairy/vendored/clipseg/__init__.py:48
↓ 2 callersFunctionget_readout_oper
(vit_features, features, use_readout, start_index=1)
imaginairy/modules/midas/midas/backbones/utils.py:126
↓ 2 callersFunctionget_reference_facial_points
Function: ---------- get reference 5 key points according to crop settings: 0. Set default crop_size: if default_
imaginairy/vendored/facexlib/detection/align_trans.py:19
↓ 2 callersMethodget_scalings
(self, sigma)
imaginairy/vendored/k_diffusion/external.py:17
↓ 2 callersMethodget_scalings
(self, sigma)
imaginairy/vendored/k_diffusion/external.py:110
↓ 2 callersMethodget_scalings
(self, sigma)
imaginairy/vendored/k_diffusion/external.py:171
↓ 2 callersFunctionget_similarity_transform
Function: ---------- Find Similarity Transform Matrix 'trans': u = src_pts[:, 0] v = src_pts[:, 1]
imaginairy/vendored/facexlib/detection/matlab_cp2tform.py:130
↓ 2 callersFunctionget_tile_coords
(d, tile_dim, overlap=0)
imaginairy/utils/feather_tile.py:78
↓ 2 callersMethodget_trainable_parameters
(self)
imaginairy/modules/sgm/autoencoding/regularizers/base.py:19
↓ 2 callersMethodget_v
(self, *args, **kwargs)
imaginairy/vendored/k_diffusion/external.py:177
← previousnext →401–500 of 2,992, ranked by callers