↓ 6 callersMethod__init__(self, in_features, hidden_features=None, out_features=None, act_layer=nn.GELU, drop=0.)
lavis/models/eva_vit.py:46
↓ 4 callersMethod__init__(
self,
model_name,
embed_dim,
image_size=224,
pool="avg",
pro
lavis/models/clip_models/timm_model.py:42
↓ 4 callersMethodinit_Qformer(cls, num_query_token, vision_width, cross_attention_freq=2, memory_bank_length=0, num_frames=0)
lavis/models/blip2_models/blip2.py:206
↓ 4 callersMethodinit_vision_encoder(
self, model_name, img_size, drop_path_rate, use_grad_checkpoint, precision
)
lavis/models/blip2_models/blip2.py:224