Skip to content

[Feature Request] Allow using TextGenerationWidget in place of ConversationalWidget when needed #521

Open
@felladrin

Description

@felladrin

Hi all,

Thank you for the ConversationalWidget addition. It's working great for my chat models. But I'd like to ask if you could add a way to set a repository to use TextGenerationWidget in place of ConversationalWidget when needed.

By the code, it doesn't seem to be possible, considering the "conversational" tag is added automatically.

$: widgetComponent =
model.pipeline_tag === "text-generation" && model.tags?.includes("conversational")
? (ConversationalWidget as typeof SvelteComponent)
: model.pipeline_tag && model.pipeline_tag in WIDGET_COMPONENTS
? WIDGET_COMPONENTS[model.pipeline_tag as keyof typeof WIDGET_COMPONENTS]
: undefined;

My use case is this base model, which has the "chat_template" because it is intended to be a foundation for chat models using ChatML, but at the same time, half of its pretraining was focused on text-completion; so it's not that good at chat, and for this reason, I'd prefer it displaying the TextGenerationWidget.


Another option would be removing the "conversational" tag from it, is there any way to do it by editing the Readme?

The current Readme is overriding the "tags",
image

but even so, the "conversational" tag is there.
image

Metadata

Metadata

Assignees

No one assigned

    Labels

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions