SYMBOL INDEX (199 symbols across 16 files)

FILE: comfyui_invsr_trimmed/inference_invsr.py
  class Namespace (line 15) | class Namespace:
    method __init__ (line 16) | def __init__(self, **kwargs):
    method __repr__ (line 20) | def __repr__(self):
  function get_configs (line 24) | def get_configs(args, log=False):

FILE: comfyui_invsr_trimmed/latent_lpips/lpips.py
  function normalize_tensor (line 11) | def normalize_tensor(in_feat,eps=1e-10):
  function spatial_average (line 15) | def spatial_average(in_tens, keepdim=True):
  function upsample (line 18) | def upsample(in_tens, out_HW=(64,64)): # assumes scale factor is same fo...
  class LPIPS (line 23) | class LPIPS(nn.Module):
    method __init__ (line 24) | def __init__(self, pretrained=True, net='alex', version='0.1', lpips=T...
    method forward (line 126) | def forward(self, in0, in1, retPerLayer=False, normalize=False):
  class ScalingLayer (line 160) | class ScalingLayer(nn.Module):
    method __init__ (line 161) | def __init__(self):
    method forward (line 166) | def forward(self, inp):
  class NetLinLayer (line 169) | class NetLinLayer(nn.Module):
    method __init__ (line 171) | def __init__(self, chn_in, chn_out=1, use_dropout=False):
    method forward (line 178) | def forward(self, x):
  class Dist2LogitLayer (line 181) | class Dist2LogitLayer(nn.Module):
    method __init__ (line 183) | def __init__(self, chn_mid=32, use_sigmoid=True):
    method forward (line 195) | def forward(self,d0,d1,eps=0.1):
  class BCERankingLoss (line 198) | class BCERankingLoss(nn.Module):
    method __init__ (line 199) | def __init__(self, chn_mid=32):
    method forward (line 206) | def forward(self, d0, d1, judge):
  function print_network (line 211) | def print_network(net):

FILE: comfyui_invsr_trimmed/latent_lpips/pretrained_networks.py
  class squeezenet (line 5) | class squeezenet(torch.nn.Module):
    method __init__ (line 6) | def __init__(self, requires_grad=False, pretrained=True):
    method forward (line 35) | def forward(self, X):
  class alexnet (line 56) | class alexnet(torch.nn.Module):
    method __init__ (line 57) | def __init__(self, requires_grad=False, pretrained=True):
    method forward (line 80) | def forward(self, X):
  class vgg16 (line 96) | class vgg16(torch.nn.Module):
    method __init__ (line 97) | def __init__(self, requires_grad=False, pretrained=True):
    method forward (line 120) | def forward(self, X):
  class vgg16_latent (line 136) | class vgg16_latent(torch.nn.Module):
    method __init__ (line 137) | def __init__(self, requires_grad=False, pretrained=True, in_chans=3):
    method forward (line 168) | def forward(self, X):
  class resnet (line 185) | class resnet(torch.nn.Module):
    method __init__ (line 186) | def __init__(self, requires_grad=False, pretrained=True, num=18):
    method forward (line 209) | def forward(self, X):

FILE: comfyui_invsr_trimmed/noise_predictor.py
  class NoisePredictor (line 23) | class NoisePredictor(ModelMixin, ConfigMixin, FromOriginalModelMixin):
    method __init__ (line 56) | def __init__(
    method _set_gradient_checkpointing (line 111) | def _set_gradient_checkpointing(self, module, value=False):
    method enable_tiling (line 115) | def enable_tiling(self, use_tiling: bool = True):
    method disable_tiling (line 123) | def disable_tiling(self):
    method enable_slicing (line 130) | def enable_slicing(self):
    method disable_slicing (line 137) | def disable_slicing(self):
    method attn_processors (line 146) | def attn_processors(self) -> Dict[str, AttentionProcessor]:
    method set_attn_processor (line 174) | def set_attn_processor(
    method set_default_attn_processor (line 211) | def set_default_attn_processor(self):
    method encode (line 233) | def encode(
    method tiled_encode (line 273) | def tiled_encode(
    method forward (line 338) | def forward(

FILE: comfyui_invsr_trimmed/pipeline_stable_diffusion_inversion_sr.py
  function retrieve_latents (line 79) | def retrieve_latents(
  function preprocess (line 92) | def preprocess(image):
  function retrieve_timesteps (line 116) | def retrieve_timesteps(
  class StableDiffusionInvEnhancePipeline (line 170) | class StableDiffusionInvEnhancePipeline(
    method __init__ (line 216) | def __init__(
    method _encode_prompt (line 309) | def _encode_prompt(
    method encode_prompt (line 342) | def encode_prompt(
    method encode_image (line 525) | def encode_image(self, image, device, num_images_per_prompt, output_hi...
    method prepare_ip_adapter_image_embeds (line 550) | def prepare_ip_adapter_image_embeds(
    method run_safety_checker (line 596) | def run_safety_checker(self, image, device, dtype):
    method decode_latents (line 611) | def decode_latents(self, latents):
    method prepare_extra_step_kwargs (line 623) | def prepare_extra_step_kwargs(self, generator, eta):
    method check_inputs (line 640) | def check_inputs(
    method get_timesteps (line 708) | def get_timesteps(self, num_inference_steps, strength, device):
    method prepare_latents (line 719) | def prepare_latents(
    method get_guidance_scale_embedding (line 789) | def get_guidance_scale_embedding(
    method guidance_scale (line 820) | def guidance_scale(self):
    method clip_skip (line 824) | def clip_skip(self):
    method do_classifier_free_guidance (line 831) | def do_classifier_free_guidance(self):
    method cross_attention_kwargs (line 835) | def cross_attention_kwargs(self):
    method num_timesteps (line 839) | def num_timesteps(self):
    method interrupt (line 843) | def interrupt(self):
    method __call__ (line 848) | def __call__(

FILE: comfyui_invsr_trimmed/sampler_invsr.py
  class BaseSampler (line 30) | class BaseSampler:
    method __init__ (line 31) | def __init__(self, configs):
    method setup_seed (line 44) | def setup_seed(self, seed=None):
    method write_log (line 51) | def write_log(self, log_str):
    method build_model (line 54) | def build_model(self):
  class InvSamplerSR (line 109) | class InvSamplerSR(BaseSampler):
    method __init__ (line 110) | def __init__(self, base_sampler):
    method sample_func (line 115) | def sample_func(self, im_cond):
    method inference (line 226) | def inference(self, image_bchw):
  function get_torch_dtype (line 229) | def get_torch_dtype(torch_dtype: str):

FILE: comfyui_invsr_trimmed/time_aware_encoder.py
  class TimeAwareEncoder (line 15) | class TimeAwareEncoder(nn.Module):
    method __init__ (line 40) | def __init__(
    method forward (line 125) | def forward(

FILE: comfyui_invsr_trimmed/utils/resize.py
  function nearest_contribution (line 32) | def nearest_contribution(x: torch.Tensor) -> torch.Tensor:
  function linear_contribution (line 38) | def linear_contribution(x: torch.Tensor) -> torch.Tensor:
  function cubic_contribution (line 45) | def cubic_contribution(x: torch.Tensor, a: float = -0.5) -> torch.Tensor:
  function gaussian_contribution (line 63) | def gaussian_contribution(x: torch.Tensor, sigma: float = 2.0) -> torch....
  function discrete_kernel (line 71) | def discrete_kernel(kernel: str, scale: float, antialiasing: bool = True...
  function reflect_padding (line 101) | def reflect_padding(x: torch.Tensor, dim: int, pad_pre: int, pad_post: i...
  function padding (line 131) | def padding(x: torch.Tensor,
  function get_padding (line 146) | def get_padding(base: torch.Tensor, kernel_size: int, x_size: int) -> ty...
  function get_weight (line 167) | def get_weight(dist: torch.Tensor,
  function reshape_tensor (line 189) | def reshape_tensor(x: torch.Tensor, dim: int, kernel_size: int) -> torch...
  function reshape_input (line 206) | def reshape_input(x: torch.Tensor) -> typing.Tuple[torch.Tensor, _I, _I,...
  function reshape_output (line 222) | def reshape_output(x: torch.Tensor, b: _I, c: _I) -> torch.Tensor:
  function cast_input (line 237) | def cast_input(x: torch.Tensor) -> typing.Tuple[torch.Tensor, _D]:
  function cast_output (line 247) | def cast_output(x: torch.Tensor, dtype: _D) -> torch.Tensor:
  function resize_1d (line 260) | def resize_1d(x: torch.Tensor,
  function downsampling_2d (line 334) | def downsampling_2d(x: torch.Tensor, k: torch.Tensor, scale: int, paddin...
  function imresize (line 354) | def imresize(x: torch.Tensor,

FILE: comfyui_invsr_trimmed/utils/util_color_fix.py
  function calc_mean_std (line 16) | def calc_mean_std(feat: Tensor, eps=1e-5):
  function adaptive_instance_normalization (line 31) | def adaptive_instance_normalization(content_feat:Tensor, style_feat:Tens...
  function wavelet_blur (line 45) | def wavelet_blur(image: Tensor, radius: int):
  function wavelet_decomposition (line 66) | def wavelet_decomposition(image: Tensor, levels=5):
  function wavelet_reconstruction (line 80) | def wavelet_reconstruction(content_feat:Tensor, style_feat:Tensor):
  function ycbcr_color_replace (line 93) | def ycbcr_color_replace(content_feat:Tensor, style_feat:Tensor):

FILE: comfyui_invsr_trimmed/utils/util_common.py
  function mkdir (line 11) | def mkdir(dir_path, delete=False, parents=True):
  function get_obj_from_str (line 21) | def get_obj_from_str(string, reload=False):
  function instantiate_from_config (line 32) | def instantiate_from_config(config):
  function str2bool (line 37) | def str2bool(v):
  function get_filenames (line 47) | def get_filenames(dir_path, exts=['png', 'jpg'], recursive=True):
  function readline_txt (line 65) | def readline_txt(txt_file):
  function scan_files_from_folder (line 74) | def scan_files_from_folder(dir_paths, exts, recursive=True):
  function write_path_to_txt (line 96) | def write_path_to_txt(

FILE: comfyui_invsr_trimmed/utils/util_ema.py
  class LitEma (line 5) | class LitEma(nn.Module):
    method __init__ (line 6) | def __init__(self, model, decay=0.9999, use_num_upates=True):
    method reset_num_updates (line 25) | def reset_num_updates(self):
    method forward (line 29) | def forward(self, model):
    method copy_to (line 50) | def copy_to(self, model):
    method store (line 64) | def store(self, parameters):
    method restore (line 73) | def restore(self, parameters):
    method resume (line 87) | def resume(self, ckpt, num_updates):

FILE: comfyui_invsr_trimmed/utils/util_image.py
  function ssim (line 14) | def ssim(img1, img2):
  function calculate_ssim (line 36) | def calculate_ssim(im1, im2, border=0, ycbcr=False):
  function calculate_psnr (line 65) | def calculate_psnr(im1, im2, border=0, ycbcr=False):
  function normalize_np (line 88) | def normalize_np(im, mean=0.5, std=0.5, reverse=False):
  function normalize_th (line 110) | def normalize_th(im, mean=0.5, std=0.5, reverse=False):
  function rgb2ycbcr (line 133) | def rgb2ycbcr(im, only_y=True):
  function rgb2ycbcrTorch (line 159) | def rgb2ycbcrTorch(im, only_y=True):
  function ycbcr2rgbTorch (line 186) | def ycbcr2rgbTorch(im):
  function bgr2rgb (line 210) | def bgr2rgb(im): return cv2.cvtColor(im, cv2.COLOR_BGR2RGB)
  function rgb2bgr (line 212) | def rgb2bgr(im): return cv2.cvtColor(im, cv2.COLOR_RGB2BGR)
  function tensor2img (line 214) | def tensor2img(tensor, rgb2bgr=True, out_type=np.uint8, min_max=(0, 1)):
  function imresize_np (line 274) | def imresize_np(img, scale, antialiasing=True):
  function calculate_weights_indices (line 346) | def calculate_weights_indices(in_length, out_length, scale, kernel, kern...
  function cubic (line 401) | def cubic(x):
  function imread (line 409) | def imread(path, chn='rgb', dtype='float32', force_gray2rgb=True, force_...
  function data_aug_np (line 455) | def data_aug_np(image, mode):
  function inverse_data_aug_np (line 502) | def inverse_data_aug_np(image, mode):
  function imshow (line 533) | def imshow(x, title=None, cbar=False):
  function imblend_with_mask (line 542) | def imblend_with_mask(im, mask, alpha=0.25):
  function imgrad (line 565) | def imgrad(im, pading_mode='mirror'):
  function convtorch (line 595) | def convtorch(im, weight, mode='reflect'):
  function random_crop (line 611) | def random_crop(im, pch_size):
  class ToTensor (line 640) | class ToTensor:
    method __init__ (line 641) | def __init__(self, max_value=1.0):
    method __call__ (line 644) | def __call__(self, im):
  class RandomCrop (line 656) | class RandomCrop:
    method __init__ (line 657) | def __init__(self, pch_size, pass_crop=False):
    method __call__ (line 661) | def __call__(self, im):
  class ImageSpliterNp (line 672) | class ImageSpliterNp:
    method __init__ (line 673) | def __init__(self, im, pch_size, stride, sf=1):
    method extract_starts (line 698) | def extract_starts(self, length):
    method __len__ (line 704) | def __len__(self):
    method __iter__ (line 707) | def __iter__(self):
    method __next__ (line 710) | def __next__(self):
    method update (line 730) | def update(self, pch_res, index_infos):
    method gather (line 745) | def gather(self):
  class ImageSpliterTh (line 749) | class ImageSpliterTh:
    method __init__ (line 750) | def __init__(self, im, pch_size, stride, sf=1, extra_bs=1, weight_type...
    method extract_starts (line 784) | def extract_starts(self, length):
    method __len__ (line 795) | def __len__(self):
    method __iter__ (line 798) | def __iter__(self):
    method __next__ (line 801) | def __next__(self):
    method update (line 826) | def update(self, pch_res, index_infos):
    method generate_kernel_1d (line 842) | def generate_kernel_1d(ksize):
    method get_weight (line 852) | def get_weight(self, height, width):
    method gather (line 865) | def gather(self):
  class Clamper (line 870) | class Clamper:
    method __init__ (line 871) | def __init__(self, min_max=(-1, 1)):
    method __call__ (line 874) | def __call__(self, im):
  class Bicubic (line 883) | class Bicubic:
    method __init__ (line 884) | def __init__(self, scale=None, out_shape=None, activate_matlab=True, r...
    method __call__ (line 890) | def __call__(self, im):
  class SmallestMaxSize (line 913) | class SmallestMaxSize:
    method __init__ (line 914) | def __init__(self, max_size, pass_resize=False, interpolation=None):
    method get_interpolation (line 926) | def get_interpolation(self, size):
    method __call__ (line 937) | def __call__(self, im):
  class SpatialAug (line 956) | class SpatialAug:
    method __init__ (line 957) | def __init__(self, pass_aug, only_hflip=False, only_vflip=False, only_...
    method __call__ (line 963) | def __call__(self, im, flag=None):

FILE: comfyui_invsr_trimmed/utils/util_net.py
  function reload_model (line 5) | def reload_model(model, ckpt):

FILE: comfyui_invsr_trimmed/utils/util_opts.py
  function update_args (line 7) | def update_args(args_json, args_parser):
  function str2bool (line 11) | def str2bool(v):

FILE: comfyui_invsr_trimmed/utils/util_sisr.py
  function modcrop (line 8) | def modcrop(im, sf):
  class Bicubic (line 15) | class Bicubic:
    method __init__ (line 16) | def __init__(self, scale=None, out_shape=None, matlab_mode=True):
    method __call__ (line 20) | def __call__(self, im):

FILE: node.py
  function split_tensor_into_batches (line 8) | def split_tensor_into_batches(tensor, batch_size):
  class LoadInvSRModels (line 44) | class LoadInvSRModels:
    method INPUT_TYPES (line 46) | def INPUT_TYPES(s):
    method loadmodel (line 61) | def loadmodel(self, sd_model, invsr_model, dtype, tiled_vae):
  class InvSRSampler (line 99) | class InvSRSampler:
    method INPUT_TYPES (line 101) | def INPUT_TYPES(s):
    method process (line 122) | def process(self, invsr_pipe, images, num_steps, cfg, batch_size, chop...