diff --git a/src/providers/replicate/handler.ts b/src/providers/replicate/handler.ts index cbe43703..060981a1 100644 --- a/src/providers/replicate/handler.ts +++ b/src/providers/replicate/handler.ts @@ -3,9 +3,13 @@ import type { HandlerPayload, Provider } from '@/types/provider' export const handlePrompt: Provider['handlePrompt'] = async(payload, signal?: AbortSignal) => { if (payload.botId === 'stable-diffusion') - return handleReplicateGenerate('db21e45d3f7023abc2a46ee38a23973f6dce16bb082a930b0c49861f96d1e5bf', payload) + return handleReplicateGenerate('ac732df83cea7fff18b8472768c88ad041fa750ff7682a21affe81863cbe77e4', payload) + if (payload.botId === 'stable-diffusion-v1') + return handleReplicateGenerate('b3d14e1cd1f9470bbb0bb68cac48e5f483e5be309551992cc33dc30654a82bb7', payload) if (payload.botId === 'waifu-diffusion') return handleReplicateGenerate('25d2f75ecda0c0bed34c806b7b70319a53a1bccad3ade1a7496524f013f48983', payload) + if (payload.botId === 'sdxl') + return handleReplicateGenerate('2a865c9a94c9992b6689365b75db2d678d5022505ed3f63a5f53929a31a46947', payload) } const handleReplicateGenerate = async(modelVersion: string, payload: HandlerPayload) => { diff --git a/src/providers/replicate/index.ts b/src/providers/replicate/index.ts index fb023226..2be1c4d1 100644 --- a/src/providers/replicate/index.ts +++ b/src/providers/replicate/index.ts @@ -17,7 +17,13 @@ const providerReplicate = () => { { id: 'stable-diffusion', type: 'image_generation', - name: 'Stable Diffusion', + name: 'Stable Diffusion 2.1', + settings: [], + }, + { + id: 'stable-diffusion-v1', + type: 'image_generation', + name: 'Stable Diffusion 1.5', settings: [], }, { @@ -26,6 +32,12 @@ const providerReplicate = () => { name: 'Waifu Diffusion', settings: [], }, + { + id: 'sdxl', + type: 'image_generation', + name: 'sdxl', + settings: [], + }, ], supportCallMethod: 'backend', handlePrompt,