feat: add upscaler & bg remover (#6967)

This commit is contained in:
darkskygit
2024-05-16 11:09:33 +00:00
parent f37bbb0784
commit a3f3d09764
7 changed files with 105 additions and 6 deletions
+10 -3
View File
@@ -29,7 +29,13 @@ export class MockCopilotTestProvider
CopilotImageToImageProvider,
CopilotImageToTextProvider
{
override readonly availableModels = ['test'];
override readonly availableModels = [
'test',
'fast-turbo-diffusion',
'lcm-sd15-i2i',
'clarity-upscaler',
'imageutils/rembg',
];
static override readonly capabilities = [
CopilotCapability.TextToText,
CopilotCapability.TextToEmbedding,
@@ -107,7 +113,7 @@ export class MockCopilotTestProvider
// ====== text to image ======
override async generateImages(
messages: PromptMessage[],
_model: string = 'test',
model: string = 'test',
_options: {
signal?: AbortSignal;
user?: string;
@@ -118,7 +124,8 @@ export class MockCopilotTestProvider
throw new Error('Prompt is required');
}
return ['https://example.com/image.jpg'];
// just let test case can easily verify the final prompt
return [`https://example.com/${model}.jpg`, prompt];
}
override async *generateImagesStream(