Python TFViTModel.from_pretrained примеры использования

Пример #1

Файл: test_modeling_tf_vit.py Проект: Kevin-Zhao-Github/oLMpics

    def test_model_from_pretrained(self):

        model = TFViTModel.from_pretrained("google/vit-base-patch16-224")
        self.assertIsNotNone(model)

Пример #2

Файл: test_modeling_tf_vision_encoder_decoder.py Проект: vuiseng9/transformers

 def get_encoder_decoder_models(self):
     encoder_model = TFViTModel.from_pretrained("google/vit-base-patch16-224-in21k", name="encoder")
     decoder_model = TFGPT2LMHeadModel.from_pretrained("../gpt2", config=self.get_decoder_config(), name="decoder")
     return {"encoder": encoder_model, "decoder": decoder_model}