Skip to content

Fix for gradio inference#1249

Closed
raivisdejus wants to merge 5 commits into
SWivid:mainfrom
raivisdejus:fix-gradio-inference
Closed

Fix for gradio inference#1249
raivisdejus wants to merge 5 commits into
SWivid:mainfrom
raivisdejus:fix-gradio-inference

Conversation

@raivisdejus
Copy link
Copy Markdown
Contributor

Fixes

  • Added F5TTS base model to the selectable model list
  • Refactored custom model selection to be able to specify relevant base model
  • Added shared community models to the custom model dropdown

Did tests:

  • For base models in English and Chinease (output seemed reasonable, but I do not understand the language)
  • Tested custom community models in languages I can understand (Latvian and Russian)

@SWivid
Copy link
Copy Markdown
Owner

SWivid commented Jan 17, 2026

Thanks.
It would be better to split it into two PRs, one with SHARED.md update, one with possible v0 base button add-back.

Personally I think the latter is not that necessary, because it has already been included in the custom options, and we want to move on without too many branches on the interface.

@raivisdejus
Copy link
Copy Markdown
Contributor Author

@SWivid Moved the SHARED.md update to another PR - #1250.

Adjusted the PR to add F5 TTS base model selection only to custom model options. I think we need it somewhere as not all available community models are listed in the SHARED.md. To use random custom fine tuned models users will need to specify the base model.

@SWivid
Copy link
Copy Markdown
Owner

SWivid commented Jan 17, 2026

@raivisdejus we already support v0 base model:

json.dumps(
dict(
dim=1024,
depth=22,
heads=16,
ff_mult=2,
text_dim=512,
text_mask_padding=False,
conv_layers=4,
pe_attn_head=1,
)
),

and the config is editable, which means you could modify the dict to any setting of a model ckpt.

@raivisdejus
Copy link
Copy Markdown
Contributor Author

Thanks for your comment, it pointed me to the right direction to implement the feature I needed in a much simpler way.

Created new PR - #1251

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants