Method __init__

eval_code/recons/models/fastmodel.py:63–69

(
            self,
            pretrained_model_name_or_path: Optional[str] = None,
        )

class G2VLM(nn.Module):
    def __init__(
            self,
            pretrained_model_name_or_path: Optional[str] = None,
        ):
        super().__init__()
        from g2vlm_utils import load_model_and_tokenizer_from_path
        self.model, self.tokenizer, self.new_token_ids,self.vit_image_transform, self.dino_transform = load_model_and_tokenizer_from_path(pretrained_model_name_or_path)


    def forward(self, images: torch.Tensor):