relbert/opt-350m-analogy

This is facebook/opt-350m fine-tuned on relbert/semeval2012_relational_similarity for analogy generation, which is to generate a word pair (eg. bird is to crow) given a query (eg. mammal is to whale) so that the query and the generated word pair form an analogy statement.

Usage

from transformers import pipeline

pipe = pipeline('text-generation', model="relbert/opt-350m-analogy")
output = pipe("mammal is to whale what")