towhee.models.coformer.coformer.create_model¶

towhee.models.coformer.coformer.create_model(model_name: Optional[str] = None, vidx_ridx: Optional[Any] = None, device: Optional[str] = None) → CoFormer[source]¶

Create CoFormer model :param model_name: CoFormer model name. :type model_name: str :param vidx_ridx: Verb index to role index. :type vidx_ridx: Any :param device: cuda or cpu. :type device: str