SYMBOL INDEX (62 symbols across 5 files)

FILE: inference.py
  function parse_args (line 15) | def parse_args():
  function init_pipeline (line 22) | def init_pipeline(config):
  function run_inference (line 33) | def run_inference(rank, world_size, config):
  function main (line 57) | def main(config):

FILE: src/tools.py
  function export_to_video (line 9) | def export_to_video(video_frames, output_video_path, fps = 12):
  function save_generation (line 22) | def save_generation(video_frames, configs, base_path, file_name=None):
  class GlobalState (line 44) | class GlobalState:
    method __init__ (line 45) | def __init__(self, state={}) -> None:
    method init_state (line 48) | def init_state(self, state={}):
    method set (line 51) | def set(self, key, value):
    method get (line 54) | def get(self, key, default=None):
  class DistController (line 58) | class DistController(object):
    method __init__ (line 59) | def __init__(self, rank, world_size, config) -> None:
    method init_dist (line 70) | def init_dist(self):
    method init_group (line 76) | def init_group(self):

FILE: src/video_crafter.py
  class VideoCrafterPipeline (line 11) | class VideoCrafterPipeline(TextToVideoSDPipeline):
    method __init__ (line 13) | def __init__(
    method __call__ (line 32) | def __call__(
    method from_pretrained (line 53) | def from_pretrained(
  class UNetVideoCrafter (line 65) | class UNetVideoCrafter(UNet3DConditionModel):
    method __init__ (line 67) | def __init__(

FILE: src/video_infinity/plugins.py
  function my_attention (line 5) | def my_attention(query, key, value, attn_mask=None, dropout_p=0.0, is_ca...
  class ModulePlugin (line 31) | class ModulePlugin:
    method __init__ (line 32) | def __init__(self, module, module_id, global_state=None):
    method is_log_node (line 40) | def is_log_node(self):
    method t (line 44) | def t(self):
    method p (line 48) | def p(self):
    method implement_forward (line 51) | def implement_forward(self):
    method set_enable (line 61) | def set_enable(self, enable=True):
    method get_new_forward (line 64) | def get_new_forward(self):
    method update_config (line 67) | def update_config(self, config:dict=None):
  class GroupNormPlugin (line 74) | class GroupNormPlugin(ModulePlugin):
    method __init__ (line 75) | def __init__(self, module, module_id, global_state=None):
    method get_new_forward (line 78) | def get_new_forward(self):
  class ConvLayerPlugin (line 106) | class ConvLayerPlugin(ModulePlugin):
    method __init__ (line 107) | def __init__(self, module, module_id, global_state=None):
    method pad_context (line 113) | def pad_context(self, h, padding=None):
    method get_new_forward (line 155) | def get_new_forward(self):
  class AttentionPlugin (line 182) | class AttentionPlugin(ModulePlugin):
    method __init__ (line 183) | def __init__(self, module, module_id, global_state=None):
    method pad_context (line 195) | def pad_context(self, h, padding=None):
    method get_topk (line 239) | def get_topk(self, q, k, v, top_k=None):
    method gather_context (line 261) | def gather_context(self, h):
    method get_new_forward (line 267) | def get_new_forward(self):
  class Conv3DPligin (line 329) | class Conv3DPligin(ModulePlugin):
    method __init__ (line 330) | def __init__(self, module, module_id, global_state=None):
    method pad_context (line 336) | def pad_context(self, h):
    method get_new_forward (line 378) | def get_new_forward(self):
  class UNetPlugin (line 387) | class UNetPlugin(ModulePlugin):
    method __init__ (line 388) | def __init__(self, module, module_id, global_state=None):
    method get_new_forward (line 391) | def get_new_forward(self):

FILE: src/video_infinity/wrapper.py
  class DistWrapper (line 5) | class DistWrapper(object):
    method __init__ (line 6) | def __init__(self, pipe, dist_controller: DistController, config) -> N...
    method switch_plugin (line 16) | def switch_plugin(self, plugin_name, enable):
    method config_plugin (line 22) | def config_plugin(self, plugin_name, config):
    method plugin_mount (line 29) | def plugin_mount(self):
    method group_norm_plugin_mount (line 41) | def group_norm_plugin_mount(self):
    method conv_plugin_mount (line 53) | def conv_plugin_mount(self):
    method conv_3d_plugin_mount (line 65) | def conv_3d_plugin_mount(self):
    method attn_plugin_mount (line 78) | def attn_plugin_mount(self):
    method unet_plugin_mount (line 90) | def unet_plugin_mount(self):
    method inference (line 97) | def inference(