Ahmadzei's picture
added 3 more tables for large emb model
5fa1a76
Here, let's also set return_full_text = False
so that output doesn't contain the prompt:
thon
torch.manual_seed(1) # doctest: +IGNORE_RESULT
prompt = """Return a list of named entities in the text.