↓ 3 callersFunction_flash_attn_forward(q, k, v, bias=None, causal=False, softmax_scale=None)
visual_instruction_tuning/llava/model/language_model/mpt/flash_attn_triton.py:329
↓ 3 callersMethod_replace_module(self, parent_module, child_name, new_module, old_module, old_module_weight = None, eval=False)
visual_instruction_tuning/peft/src/peft/tuners/dora.py:406
↓ 3 callersFunctionget_loader(args, split='train', mode='train',
batch_size=32, workers=4, distributed=False, gpu=0,
image_video_text_understanding/VL-T5/src/nlvr_clip_data.py:254
↓ 3 callersFunctionget_loader(args, split='train', mode='train',
batch_size=32, workers=4, distributed=False, gpu=0,
image_video_text_understanding/VL-T5/src/refcoco_data.py:349
↓ 3 callersFunctionget_loader(args, split='train', mode='train',
batch_size=32, workers=4, distributed=False, gpu=0,
image_video_text_understanding/VL-T5/src/caption_clip_data.py:345
↓ 3 callersFunctionget_loader(args, split='train', mode='train',
batch_size=32, workers=4, distributed=False, gpu=0,
image_video_text_understanding/VL-T5/src/vcr_data.py:398