SYMBOL INDEX (99 symbols across 10 files)

FILE: gradio_app.py
  function pad_np_bgr_image (line 69) | def pad_np_bgr_image(np_image, scale=1.25):
  function process_faceid_image (line 79) | def process_faceid_image(pil_faceid_image):
  function prepare_single_faceid_cond_kwargs (line 106) | def prepare_single_faceid_cond_kwargs(pil_faceid_image=None, pil_faceid_...
  function text_to_single_id_generation_process (line 140) | def text_to_single_id_generation_process(
  function text_to_multi_id_generation_process (line 190) | def text_to_multi_id_generation_process(
  function image_to_single_id_generation_process (line 253) | def image_to_single_id_generation_process(
  function text_to_single_id_generation_block (line 320) | def text_to_single_id_generation_block():
  function text_to_multi_id_generation_block (line 389) | def text_to_multi_id_generation_block():
  function image_to_single_id_generation_block (line 477) | def image_to_single_id_generation_block():

FILE: uniportrait/curricular_face/backbone/__init__.py
  function get_model (line 24) | def get_model(key):

FILE: uniportrait/curricular_face/backbone/common.py
  function initialize_weights (line 8) | def initialize_weights(modules):
  class Flatten (line 27) | class Flatten(Module):
    method forward (line 31) | def forward(self, input):
  class SEModule (line 35) | class SEModule(Module):
    method __init__ (line 39) | def __init__(self, channels, reduction):
    method forward (line 61) | def forward(self, x):

FILE: uniportrait/curricular_face/backbone/model_irse.py
  class BasicBlockIR (line 11) | class BasicBlockIR(Module):
    method __init__ (line 15) | def __init__(self, in_channel, depth, stride):
    method forward (line 30) | def forward(self, x):
  class BottleneckIR (line 37) | class BottleneckIR(Module):
    method __init__ (line 41) | def __init__(self, in_channel, depth, stride):
    method forward (line 64) | def forward(self, x):
  class BasicBlockIRSE (line 71) | class BasicBlockIRSE(BasicBlockIR):
    method __init__ (line 73) | def __init__(self, in_channel, depth, stride):
  class BottleneckIRSE (line 78) | class BottleneckIRSE(BottleneckIR):
    method __init__ (line 80) | def __init__(self, in_channel, depth, stride):
  class Bottleneck (line 85) | class Bottleneck(namedtuple('Block', ['in_channel', 'depth', 'stride'])):
  function get_block (line 89) | def get_block(in_channel, depth, num_units, stride=2):
  function get_blocks (line 94) | def get_blocks(num_layers):
  class Backbone (line 141) | class Backbone(Module):
    method __init__ (line 143) | def __init__(self, input_size, num_layers, mode='ir'):
    method forward (line 200) | def forward(self, x, return_mid_feats=False):
  function IR_18 (line 216) | def IR_18(input_size):
  function IR_34 (line 224) | def IR_34(input_size):
  function IR_50 (line 232) | def IR_50(input_size):
  function IR_101 (line 240) | def IR_101(input_size):
  function IR_152 (line 248) | def IR_152(input_size):
  function IR_200 (line 256) | def IR_200(input_size):
  function IR_SE_50 (line 264) | def IR_SE_50(input_size):
  function IR_SE_101 (line 272) | def IR_SE_101(input_size):
  function IR_SE_152 (line 280) | def IR_SE_152(input_size):
  function IR_SE_200 (line 288) | def IR_SE_200(input_size):

FILE: uniportrait/curricular_face/backbone/model_resnet.py
  function conv3x3 (line 10) | def conv3x3(in_planes, out_planes, stride=1):
  function conv1x1 (line 22) | def conv1x1(in_planes, out_planes, stride=1):
  class Bottleneck (line 29) | class Bottleneck(Module):
    method __init__ (line 32) | def __init__(self, inplanes, planes, stride=1, downsample=None):
    method forward (line 44) | def forward(self, x):
  class ResNet (line 67) | class ResNet(Module):
    method __init__ (line 71) | def __init__(self, input_size, block, layers, zero_init_residual=True):
    method _make_layer (line 105) | def _make_layer(self, block, planes, blocks, stride=1):
    method forward (line 121) | def forward(self, x):
  function ResNet_50 (line 141) | def ResNet_50(input_size, **kwargs):
  function ResNet_101 (line 149) | def ResNet_101(input_size, **kwargs):
  function ResNet_152 (line 157) | def ResNet_152(input_size, **kwargs):

FILE: uniportrait/curricular_face/inference.py
  function inference (line 13) | def inference(name, weight, src_norm_dir):

FILE: uniportrait/inversion.py
  function _encode_text_with_negative (line 16) | def _encode_text_with_negative(model: StableDiffusionPipeline, prompt: s...
  function _encode_image (line 24) | def _encode_image(model: StableDiffusionPipeline, image: np.ndarray) -> T:
  function _next_step (line 33) | def _next_step(model: StableDiffusionPipeline, model_output: T, timestep...
  function _get_noise_pred (line 46) | def _get_noise_pred(model: StableDiffusionPipeline, latent: T, t: T, con...
  function _ddim_loop (line 55) | def _ddim_loop(model: StableDiffusionPipeline, z0, prompt, guidance_scal...
  function make_inversion_callback (line 69) | def make_inversion_callback(zts, offset: int = 0) -> [T, InversionCallba...
  function ddim_inversion (line 80) | def ddim_inversion(model: StableDiffusionPipeline, x0: np.ndarray, promp...

FILE: uniportrait/resampler.py
  function FeedForward (line 11) | def FeedForward(dim, mult=4):
  function reshape_tensor (line 21) | def reshape_tensor(x, heads):
  class PerceiverAttention (line 32) | class PerceiverAttention(nn.Module):
    method __init__ (line 33) | def __init__(self, *, dim, dim_head=64, heads=8):
    method forward (line 47) | def forward(self, x, latents, attention_mask=None):
  class UniPortraitFaceIDResampler (line 89) | class UniPortraitFaceIDResampler(torch.nn.Module):
    method __init__ (line 90) | def __init__(
    method forward (line 132) | def forward(

FILE: uniportrait/uniportrait_attention_processor.py
  class AttentionArgs (line 8) | class AttentionArgs(object):
    method __init__ (line 9) | def __init__(self) -> None:
    method reset (line 27) | def reset(self):
    method __repr__ (line 45) | def __repr__(self):
  function expand_first (line 54) | def expand_first(feat, scale=1., ):
  function concat_first (line 65) | def concat_first(feat, dim=2, scale=1.):
  function calc_mean_std (line 70) | def calc_mean_std(feat, eps: float = 1e-5):
  function adain (line 76) | def adain(feat):
  class UniPortraitLoRAAttnProcessor2_0 (line 85) | class UniPortraitLoRAAttnProcessor2_0(nn.Module):
    method __init__ (line 87) | def __init__(
    method __call__ (line 106) | def __call__(
  class UniPortraitLoRAIPAttnProcessor2_0 (line 206) | class UniPortraitLoRAIPAttnProcessor2_0(nn.Module):
    method __init__ (line 208) | def __init__(self, hidden_size, cross_attention_dim=None, rank=128, ne...
    method __call__ (line 243) | def __call__(
  class UniPortraitCNAttnProcessor2_0 (line 422) | class UniPortraitCNAttnProcessor2_0:
    method __init__ (line 423) | def __init__(self, num_ip_tokens=4, num_faceid_tokens=16):
    method __call__ (line 428) | def __call__(

FILE: uniportrait/uniportrait_pipeline.py
  class ImageProjModel (line 15) | class ImageProjModel(nn.Module):
    method __init__ (line 18) | def __init__(self, cross_attention_dim=1024, clip_embeddings_dim=1024,...
    method forward (line 26) | def forward(self, image_embeds):
  class UniPortraitPipeline (line 34) | class UniPortraitPipeline:
    method __init__ (line 36) | def __init__(self, sd_pipe, image_encoder_path, ip_ckpt=None, face_bac...
    method init_image_proj (line 73) | def init_image_proj(self):
    method init_faceid_proj (line 81) | def init_faceid_proj(self):
    method set_uniportrait_and_ip_adapter (line 92) | def set_uniportrait_and_ip_adapter(self):
    method load_uniportrait_and_ip_adapter (line 139) | def load_uniportrait_and_ip_adapter(self):
    method get_ip_embeds (line 167) | def get_ip_embeds(self, pil_ip_image):
    method get_single_faceid_embeds (line 176) | def get_single_faceid_embeds(self, pil_face_images, face_structure_sca...
    method generate (line 220) | def generate(