def METHOD_NAME(self):
    subfolder = "bert"
    model_id = "hf-internal-testing/tiny-random-bert-sharded-subfolder"
    with self.assertRaises(OSError):
        _ = FlaxBertModel.from_pretrained(model_id)
    model = FlaxBertModel.from_pretrained(model_id, subfolder=subfolder)
    self.assertIsNotNone(model)