add more Replicate models #1102
Conversation
There was a problem hiding this comment.
awesome ❤️
one thing to take into account is that each HF model page is (automatically or manually) identified as one pipeline type aka. task type, and then the API schema is conditioned by this pipeline type
I haven't looked in depth, but some of the Replicate endpoints might very well not have an API that's conforming to the HF pipeline type. In those cases we should exclude them from the list because the widget won't work.
cc @SBrandeis for more details:)
zeke
force-pushed
the
add-more-mapping-to-replicate-models
branch
from
676d45d to
e676cc7
Compare
|
Thanks @SBrandeis. I rebased and added ids to all the versioned models. For posterity, here's the scrappy bit of code I wrote to grab those versions: https://github.com/zeke/get-replicate-model-versions/blob/c33c995cead6df4ed978930b71e0c8c7c3a5941f/index.js#L62-L81 |
|
The model list in this PR is too amibitious, so I'm gonna close it. Too many modalities to try to support all at once. Instead we can approach this incrementally. Here's a new PR that adds a handful more models, with tests: #1130 |
This PR adds mappings for more Replicate models, along with tests for each new model. ``` vitest run --config vitest.config.mts -t "Replicate" ... ✓ Replicate (9) 8666ms ✓ textToImage canonical - black-forest-labs/FLUX.1-schnell 1214ms ✓ textToImage canonical - black-forest-labs/FLUX.1-dev 3472ms ✓ textToImage canonical - stabilityai/stable-diffusion-3.5-large-turbo 1460ms ✓ textToImage versioned - ByteDance/SDXL-Lightning 2612ms ✓ textToImage versioned - ByteDance/Hyper-SD 4346ms ✓ textToImage versioned - playgroundai/playground-v2.5-1024px-aesthetic 4099ms ✓ textToImage versioned - stabilityai/stable-diffusion-xl-base-1.0 7205ms ↓ textToSpeech versioned [skipped] ↓ textToSpeech OuteTTS [skipped] ``` Replaces #1102 cc @julien-c @SBrandeis --------- Co-authored-by: SBrandeis <[email protected]> Co-authored-by: Simon Brandeis <[email protected]>
This is a work in progress PR to expand the list of Replicate models.
Every model in this list has over 10,000 runs on Replicate.
To Do