Python GPTJModel.from_pretrained Examples

Example #1

 def test_model_from_pretrained(self):
     for model_name in GPTJ_PRETRAINED_MODEL_ARCHIVE_LIST[:1]:
         model = GPTJModel.from_pretrained(model_name,
                                           revision="float16",
                                           torch_dtype=torch.float16)
         self.assertIsNotNone(model)

Example #2

File: test_modeling_gptj.py Project: ruyimarone/pytorch-pretrained-BERT

 def test_model_from_pretrained(self):
     for model_name in GPTJ_PRETRAINED_MODEL_ARCHIVE_LIST[:1]:
         model = GPTJModel.from_pretrained(model_name)
         self.assertIsNotNone(model)