MCPcopy Create free account

hub / github.com/brycedrennan/imaginAIry / functions

Functions2,992 in github.com/brycedrennan/imaginAIry

↓ 1 callersMethodpredict_eps_from_z_and_v
(self, x_t, t, v)
imaginairy/modules/diffusion/ddpm.py:451
↓ 1 callersMethodpredict_start_from_z_and_v
(self, x_t, t, v)
imaginairy/modules/diffusion/ddpm.py:443
↓ 1 callersMethodpreprocess
(self, x)
imaginairy/modules/clip_embedders.py:118
↓ 1 callersMethodpreprocess
(self, x)
imaginairy/modules/sgm/encoders/modules.py:647
↓ 1 callersFunctionprompt_library_filepath
(library_path)
imaginairy/enhancers/prompt_expansion.py:41
↓ 1 callersFunctionprompt_mutator
Given a prompt and a list of kwarg schedules, return a series of prompts that follow the schedule. kwarg_schedules example: { "p
imaginairy/utils/prompt_schedules.py:48
↓ 1 callersMethodpropose_step
(self, error)
imaginairy/vendored/k_diffusion/sampling.py:458
↓ 1 callersMethodprune_heads
(self, heads)
imaginairy/vendored/blip/nlvr_encoder.py:301
↓ 1 callersMethodprune_heads
(self, heads)
imaginairy/vendored/blip/med.py:282
↓ 1 callersMethodpt2np
(self, x)
imaginairy/modules/midas/utils.py:17
↓ 1 callersMethodq_sample
(self, x_start, t, noise=None)
imaginairy/modules/diffusion/upscaling.py:67
↓ 1 callersMethodq_sample
(self, x_start, t, noise=None)
imaginairy/modules/sgm/encoders/modules.py:912
↓ 1 callersFunctionrank
(image_features, text_features, top_count=100)
imaginairy/enhancers/describe_image_clip.py:49
↓ 1 callersMethodrank_answer
(self, question_states, question_atts, answer_ids, answer_atts, k)
imaginairy/vendored/blip/blip_vqa.py:163
↓ 1 callersFunctionrealesrgan_upsampler
(tile=512, tile_pad=50, ultrasharp=False)
imaginairy/enhancers/upscale_realesrgan.py:15
↓ 1 callersMethodregister_schedule
( self, beta_schedule="linear", timesteps=1000, linear_start=1e-4, lin
imaginairy/modules/diffusion/upscaling.py:19
↓ 1 callersMethodregister_schedule
( self, beta_schedule="linear", timesteps=1000, linear_start=1e-4, lin
imaginairy/modules/sgm/encoders/modules.py:864
↓ 1 callersMethodremove_noise
(self, x: Tensor, noise: Tensor, step: int)
imaginairy/vendored/refiners/foundationals/latent_diffusion/schedulers/scheduler.py:121
↓ 1 callersFunctionremove_pad
(x)
imaginairy/img_processors/densepose.py:56
↓ 1 callersFunctionremove_pkg_resources_requirement
(script_path)
imaginairy/cli/unslow_the_cli.py:30
↓ 1 callersFunctionreplace_color
(target_img, color_src_img)
imaginairy/api/colorize.py:63
↓ 1 callersFunctionreplace_value_at_path
Replace a value in a nested dictionary using a path.
imaginairy/utils/data_distorter.py:167
↓ 1 callersFunctionresample_fine_and_coarse_segm_tensors_to_bbox
Resample fine and coarse segmentation tensors to the given bounding box and derive labels for each pixel of the bounding box Args:
imaginairy/img_processors/densepose.py:591
↓ 1 callersFunctionresample_uv_tensors_to_bbox
Resamples U and V coordinate estimates for the given bounding box Args: u (tensor [1, C, H, W] of float): U coordinates v (t
imaginairy/img_processors/densepose.py:626
↓ 1 callersMethodrescaled_pos_emb
(self, new_size)
imaginairy/vendored/clipseg/__init__.py:184
↓ 1 callersMethodreset_num_updates
(self)
imaginairy/modules/ema.py:33
↓ 1 callersFunctionresize_image_with_pad_torch
( img, resolution, upscale_method="bicubic", mode="constant" )
imaginairy/img_processors/densepose.py:34
↓ 1 callersFunctionresnet50_backbone
Constructs a ResNet-50 model_hyper.
imaginairy/vendored/facexlib/assessment/hyperiqa_net.py:235
↓ 1 callersFunctionsafer_memory
(x)
imaginairy/img_processors/densepose.py:25
↓ 1 callersFunctionsafety_filter
(x)
imaginairy/api/video_sample.py:417
↓ 1 callersFunctionsafety_models
()
imaginairy/utils/safety.py:119
↓ 1 callersMethodsample
(self)
imaginairy/modules/distributions.py:20
↓ 1 callersMethodsample
( self, cond: Dict, uc: Union[Dict, None] = None, batch_size: int = 16,
imaginairy/modules/sgm/diffusion.py:255
↓ 1 callersMethodsample
(self)
imaginairy/modules/sgm/distributions/distributions.py:19
↓ 1 callersMethodsample
(self)
imaginairy/modules/sgm/distributions/distributions.py:39
↓ 1 callersMethodsample_noise_schedule
(self)
imaginairy/vendored/refiners/foundationals/latent_diffusion/schedulers/scheduler.py:98
↓ 1 callersMethodsampler_step
( self, old_denoised, previous_sigma, sigma, next_sigma, denoi
imaginairy/modules/sgm/diffusionmodules/sampling.py:315
↓ 1 callersFunctionsaturation
(image)
imaginairy/vendored/smart_crop.py:14
↓ 1 callersMethodsave_attention_map
(self, attention_map)
imaginairy/vendored/blip/vit.py:76
↓ 1 callersMethodsave_attention_map
(self, attention_map)
imaginairy/vendored/blip/nlvr_encoder.py:129
↓ 1 callersMethodsave_attention_map
(self, attention_map)
imaginairy/vendored/blip/med.py:139
↓ 1 callersMethodsave_image_as_base64
(image: "Image.Image")
imaginairy/schema.py:194
↓ 1 callersFunctionsave_lora_patterns
()
imaginairy/weight_management/generate_weight_info.py:77
↓ 1 callersFunctionsave_patterns
()
imaginairy/weight_management/generate_weight_info.py:129
↓ 1 callersFunctionsave_to_safetensors
(path: Path | str, tensors: dict[str, Tensor], metadata: dict[str, str] | None = None)
imaginairy/vendored/refiners/fluxion/utils.py:186
↓ 1 callersFunctionsave_video_bounce
( samples: torch.Tensor, video_filename: str, output_fps: int, interpolate_fps=60 )
imaginairy/api/video_sample.py:324
↓ 1 callersMethodscale
(self)
imaginairy/vendored/refiners/fluxion/adapters/lora.py:48
↓ 1 callersMethodscale_model_input
For compatibility with schedulers that need to scale the input according to the current timestep.
imaginairy/vendored/refiners/foundationals/latent_diffusion/schedulers/scheduler.py:80
↓ 1 callersMethodscore
(self, target_image, crop)
imaginairy/vendored/smart_crop.py:337
↓ 1 callersFunctionsegmodel
()
imaginairy/img_processors/segformer_b2_clothes.py:133
↓ 1 callersFunctionselect_images_by_duration_at_fps
select the proper image to show for each frame of a video.
imaginairy/utils/animations.py:155
↓ 1 callersMethodset_clip_text_embedding
(self, clip_text_embedding: Tensor)
imaginairy/vendored/refiners/foundationals/latent_diffusion/stable_diffusion_xl/unet.py:275
↓ 1 callersMethodset_clip_text_embedding
(self, clip_text_embedding: Tensor)
imaginairy/vendored/refiners/foundationals/latent_diffusion/stable_diffusion_1/unet.py:284
↓ 1 callersMethodset_control_models
(self, control_models)
imaginairy/modules/cldm.py:384
↓ 1 callersMethodset_controlnet_condition
(self, condition: Tensor)
imaginairy/vendored/refiners/foundationals/latent_diffusion/reference_only_control.py:139
↓ 1 callersMethodset_dense_positional_embedding
(self, dense_positional_embedding: Tensor)
imaginairy/vendored/refiners/foundationals/segment_anything/mask_decoder.py:262
↓ 1 callersMethodset_image_embedding
(self, image_embedding: Tensor)
imaginairy/vendored/refiners/foundationals/segment_anything/mask_decoder.py:250
↓ 1 callersMethodset_image_prompt
( self, images: list[Image.Image], scale: float, model_type: str = "normal" )
imaginairy/modules/refiners_sd.py:162
↓ 1 callersMethodset_inference_steps
(self, num_steps: int, first_step: int = 0)
imaginairy/vendored/refiners/foundationals/latent_diffusion/model.py:35
↓ 1 callersMethodset_inpainting_conditions
( self, target_image: Image.Image, mask: Image.Image, latents_size: tuple[int,
imaginairy/modules/refiners_sd.py:590
↓ 1 callersMethodset_inpainting_conditions
( self, target_image: Image.Image, mask: Image.Image, latents_size: tuple[int,
imaginairy/vendored/refiners/foundationals/latent_diffusion/stable_diffusion_1/model.py:137
↓ 1 callersMethodset_mask_embedding
(self, mask_embedding: Tensor)
imaginairy/vendored/refiners/foundationals/segment_anything/mask_decoder.py:258
↓ 1 callersMethodset_point_embedding
(self, point_embedding: Tensor)
imaginairy/vendored/refiners/foundationals/segment_anything/mask_decoder.py:254
↓ 1 callersMethodset_pooled_text_embedding
(self, pooled_text_embedding: Tensor)
imaginairy/vendored/refiners/foundationals/latent_diffusion/stable_diffusion_xl/unet.py:284
↓ 1 callersMethodset_scale
(self, scale: float)
imaginairy/vendored/refiners/foundationals/latent_diffusion/stable_diffusion_1/controlnet.py:172
↓ 1 callersMethodset_self_attention_guidance
(self, enable: bool, scale: float = 1.0)
imaginairy/vendored/refiners/foundationals/latent_diffusion/model.py:77
↓ 1 callersMethodset_tile_mode
For creating seamless tile images. Args: tile_mode: One of "", "x", "y", "xy". If "x", the image will be tiled horizonta
imaginairy/modules/refiners_sd.py:97
↓ 1 callersMethodset_time_ids
(self, time_ids: Tensor)
imaginairy/vendored/refiners/foundationals/latent_diffusion/stable_diffusion_xl/unet.py:281
↓ 1 callersMethodset_timestep
(self, timestep: Tensor)
imaginairy/vendored/refiners/foundationals/latent_diffusion/stable_diffusion_xl/unet.py:278
↓ 1 callersMethodset_timestep
(self, timestep: Tensor)
imaginairy/vendored/refiners/foundationals/latent_diffusion/stable_diffusion_1/unet.py:287
↓ 1 callersMethodset_type_mask
(self, type_mask: Int[Tensor, "1 num_points"])
imaginairy/vendored/refiners/foundationals/segment_anything/prompt_encoder.py:92
↓ 1 callersMethodset_unet_context
( self, *, timestep: Tensor, clip_text_embedding: Tensor, pooled_text_
imaginairy/vendored/refiners/foundationals/latent_diffusion/stable_diffusion_xl/model.py:65
↓ 1 callersMethodshared_step
(self, batch: Dict)
imaginairy/modules/sgm/diffusion.py:176
↓ 1 callersMethodshorten_tree_repr
Shorten the tree representation to a given number of lines around a given line index.
imaginairy/vendored/refiners/fluxion/layers/module.py:193
↓ 1 callersFunctionshrink_list
(items, max_size)
imaginairy/utils/__init__.py:220
↓ 1 callersFunctionshuffle_map_np
(img: "np.ndarray", h=None, w=None, f=256)
imaginairy/img_processors/control_modes.py:174
↓ 1 callersMethodsliced_encode
Encodes the image in slices (for lower memory usage).
imaginairy/modules/refiners_sd.py:688
↓ 1 callersMethodsoftmax_temperature
(tensor, temperature)
imaginairy/vendored/facexlib/headpose/hopenet_arch.py:45
↓ 1 callersFunctionspatial_average
(x, keepdim=True)
imaginairy/modules/sgm/autoencoding/lpips/loss/lpips.py:150
↓ 1 callersFunctionsquare_bbox
Adjusts the bounding box to make it as close to a square as possible while ensuring it does not exceed the max_size of the image and still in
imaginairy/img_processors/densepose.py:128
↓ 1 callersMethodstart
(self)
imaginairy/utils/memory_tracker.py:23
↓ 1 callersMethodstart
(self)
imaginairy/utils/log_utils.py:227
↓ 1 callersMethodstep
Updates the step count.
imaginairy/vendored/k_diffusion/utils.py:156
↓ 1 callersMethodstop
(self)
imaginairy/utils/memory_tracker.py:30
↓ 1 callersMethodstop
(self)
imaginairy/utils/log_utils.py:233
↓ 1 callersFunctionstructural_copy
(m: T)
imaginairy/vendored/refiners/fluxion/layers/chain.py:113
↓ 1 callersMethodstructural_copy
(self: TLatentDiffusionModel)
imaginairy/modules/refiners_sd.py:385
↓ 1 callersFunctionsummarize_tensor
(tensor: torch.Tensor, /)
imaginairy/vendored/refiners/fluxion/utils.py:190
↓ 1 callersFunctionsurprise_me_prompts
( img, person=None, width=None, height=None, steps=30, seed=None, use_controlnet=True )
imaginairy/utils/surprise_me.py:143
↓ 1 callersFunctiont_fn_a
(sigma)
tests/samplers/test_base.py:11
↓ 1 callersFunctiont_fn_b
(sigma)
tests/samplers/test_base.py:14
↓ 1 callersMethodt_to_sigma
(self, t)
imaginairy/vendored/k_diffusion/external.py:94
↓ 1 callersFunctiontensor_to_image
Convert a Tensor to a PIL Image. The tensor must have shape `[1, channels, height, width]` where the number of channels is either 1 (gra
imaginairy/vendored/refiners/fluxion/utils.py:137
↓ 1 callersMethodtext_transformer_forward
(self, x: torch.Tensor, attn_mask=None)
imaginairy/modules/encoders.py:219
↓ 1 callersMethodtext_transformer_forward
(self, x: torch.Tensor, attn_mask=None)
imaginairy/modules/sgm/encoders/modules.py:504
↓ 1 callersMethodtext_transformer_forward
(self, x: torch.Tensor, attn_mask=None)
imaginairy/modules/sgm/encoders/modules.py:579
↓ 1 callersFunctiontforminv
Function: ---------- apply the inverse of affine transform 'trans' to uv Parameters: ---------- @trans: 3x3 np.array
imaginairy/vendored/facexlib/detection/matlab_cp2tform.py:37
↓ 1 callersFunctiontie_encoder_decoder_weights
( encoder: nn.Module, decoder: nn.Module, base_model_prefix: str, skip_key: str )
imaginairy/vendored/blip/blip_pretrain.py:334
↓ 1 callersFunctiontie_encoder_to_decoder_recursively
( decoder_pointer: nn.Module, encoder_pointer: nn.Module, module_name: str, un
imaginairy/vendored/blip/blip_pretrain.py:343
↓ 1 callersFunctiontile_process
Process an image by tiling it, processing each tile, and then merging them back into one image. Args: img (Tensor): The input image tens
imaginairy/utils/tile_up.py:12
← previousnext →1,101–1,200 of 2,992, ranked by callers