SYMBOL INDEX (47 symbols across 5 files)

FILE: eval_overall_mps_on_hpdv2.py
  function infer_one_sample (line 33) | def infer_one_sample(image, prompt, clip_model, clip_processor, tokenize...
  function infer_example (line 80) | def infer_example(images, prompt, clip_model, clip_processor, tokenizer,...
  function acc (line 89) | def acc(score_sample, predict_sample):
  function inversion_score (line 111) | def inversion_score(predict_sample, score_sample):
  function main (line 122) | def main():

FILE: eval_overall_mps_on_imagereward.py
  function infer_one_sample (line 34) | def infer_one_sample(image, prompt, clip_model, clip_processor, tokenize...
  function infer_example (line 80) | def infer_example(images, prompt, clip_model, clip_processor, tokenizer,...
  function acc (line 90) | def acc(score_sample, predict_sample):
  function main (line 113) | def main():

FILE: trainer/models/base_model.py
  class BaseModelConfig (line 6) | class BaseModelConfig:

FILE: trainer/models/clip_model.py
  class XCLIPModel (line 17) | class XCLIPModel(HFCLIPModel):
    method __init__ (line 18) | def __init__(self, config: CLIPConfig):
    method get_text_features (line 21) | def get_text_features(
    method get_image_features (line 61) | def get_image_features(
  class ClipModelConfig (line 93) | class ClipModelConfig(BaseModelConfig):
  class CLIPModel (line 98) | class CLIPModel(nn.Module):
    method __init__ (line 99) | def __init__(self, ckpt):
    method get_text_features (line 104) | def get_text_features(self, *args, **kwargs):
    method get_image_features (line 107) | def get_image_features(self, *args, **kwargs):
    method forward (line 110) | def forward(self, text_inputs=None, image_inputs=None, condition_input...
    method logit_scale (line 135) | def logit_scale(self):
    method save (line 138) | def save(self, path):

FILE: trainer/models/cross_modeling.py
  function exists (line 8) | def exists(val):
  function default (line 11) | def default(val, d):
  class LayerNorm (line 18) | class LayerNorm(nn.Module):
    method __init__ (line 19) | def __init__(self, dim):
    method forward (line 24) | def forward(self, x):
  class Residual (line 30) | class Residual(nn.Module):
    method __init__ (line 31) | def __init__(self, fn):
    method forward (line 35) | def forward(self, x, *args, **kwargs):
  class RotaryEmbedding (line 43) | class RotaryEmbedding(nn.Module):
    method __init__ (line 44) | def __init__(self, dim):
    method forward (line 49) | def forward(self, max_seq_len, *, device):
  function rotate_half (line 55) | def rotate_half(x):
  function apply_rotary_pos_emb (line 61) | def apply_rotary_pos_emb(pos, t):
  class SwiGLU (line 69) | class SwiGLU(nn.Module):
    method forward (line 70) | def forward(self, x):
  class ParallelTransformerBlock (line 78) | class ParallelTransformerBlock(nn.Module):
    method __init__ (line 79) | def __init__(self, dim, dim_head=64, heads=8, ff_mult=4):
    method get_rotary_embedding (line 102) | def get_rotary_embedding(self, n, device):
    method forward (line 110) | def forward(self, x, attn_mask=None):
  class CrossAttention (line 172) | class CrossAttention(nn.Module):
    method __init__ (line 173) | def __init__(
    method forward (line 207) | def forward(self, x, context, mask):
  class Cross_model (line 261) | class Cross_model(nn.Module):
    method __init__ (line 262) | def __init__(
    method forward (line 281) | def forward(