SYMBOL INDEX (41 symbols across 5 files)

FILE: dataset_lib/dataset_eval_MoMA.py
  function Dataset_evaluate_MoMA (line 8) | def Dataset_evaluate_MoMA(rgb_path, prompt,subject, mask_path, moMA_main...

FILE: model_lib/attention_processor.py
  function get_mask_from_cross (line 10) | def get_mask_from_cross(attn_processors):
  class IPAttnProcessor (line 21) | class IPAttnProcessor(nn.Module):
    method __init__ (line 35) | def __init__(self, hidden_size, cross_attention_dim=None, scale=1.0, n...
    method __call__ (line 54) | def __call__(
  class IPAttnProcessor_Self (line 137) | class IPAttnProcessor_Self(nn.Module):
    method __init__ (line 151) | def __init__(self, hidden_size, cross_attention_dim=None, scale=1.0, n...
    method __call__ (line 170) | def __call__(

FILE: model_lib/moMA_generator.py
  function get_subject_idx (line 10) | def get_subject_idx(model,prompt,src_subject,device):
  function add_function (line 20) | def add_function(model):
  class ImageProjModel (line 66) | class ImageProjModel(torch.nn.Module):
    method __init__ (line 68) | def __init__(self, cross_attention_dim=1024, clip_embeddings_dim=1024,...
    method forward (line 76) | def forward(self, image_embeds):
  class MoMA_generator (line 83) | class MoMA_generator:
    method __init__ (line 84) | def __init__(self, device,args):
    method init_proj (line 110) | def init_proj(self):
    method set_ip_adapter (line 118) | def set_ip_adapter(self):
    method get_image_embeds_CFG (line 138) | def get_image_embeds_CFG(self, llava_emb):
    method get_image_crossAttn_feature (line 144) | def get_image_crossAttn_feature(
    method get_image_selfAttn_feature (line 158) | def get_image_selfAttn_feature(
    method generate_with_MoMA (line 184) | def generate_with_MoMA(
    method set_selfAttn_strength (line 229) | def set_selfAttn_strength(self, strength):
    method set_cross_subject_idxs (line 236) | def set_cross_subject_idxs(self, subject_idxs):
    method set_self_mask (line 241) | def set_self_mask(self,mode,id_ig='', mask=None): #only have effect on...
    method toggle_enable_flag (line 256) | def toggle_enable_flag(self, processor_enable_mode):
    method toggle_extract_inject_flag (line 269) | def toggle_extract_inject_flag(self, processor_name, mode): # mode: st...
    method reset_all (line 278) | def reset_all(self,keep_self=False):

FILE: model_lib/modules.py
  function add_function (line 18) | def add_function(model):
  class LlamaMLP_mapping (line 51) | class LlamaMLP_mapping(nn.Module):
    method __init__ (line 52) | def __init__(self, hidden_size,hidden_size_out):
    method forward (line 62) | def forward(self, x):
    method init_linear (line 66) | def init_linear(self):
  class MoMA_main_modal (line 74) | class MoMA_main_modal(nn.Module):
    method __init__ (line 75) | def __init__(self,args):
    method load_saved_components (line 94) | def load_saved_components(self):
    method freeze_modules (line 108) | def freeze_modules(self):
    method forward_MLLM (line 114) | def forward_MLLM(self,batch):
    method reset (line 136) | def reset(self):
    method generate_images (line 139) | def generate_images(self, rgb_path, mask_path, subject, prompt, streng...

FILE: model_lib/utils.py
  function parse_args (line 6) | def parse_args():
  function show_PIL_image (line 16) | def show_PIL_image(tensor):