feat: add lora support (#6977)

This commit is contained in:
darkskygit
2024-05-20 05:05:33 +00:00
parent 53ee1801e6
commit f2866f57c9
3 changed files with 118 additions and 13 deletions

View File

@@ -0,0 +1,13 @@
import { PrismaClient } from '@prisma/client';
import { refreshPrompts } from './utils/prompts';
export class UpdatePrompts1715936358947 {
// do the migration
static async up(db: PrismaClient) {
await refreshPrompts(db);
}
// revert the migration
static async down(_db: PrismaClient) {}
}

View File

@@ -83,6 +83,68 @@ export const prompts: Prompt[] = [
model: 'imageutils/rembg',
messages: [],
},
{
name: 'debug:action:fal-sdturbo-clay',
action: 'image',
model: 'fast-turbo-diffusion',
messages: [
{
role: 'user',
content: 'claymation, clay, {{content}}',
params: {
lora: [
'https://models.affine.pro/fal/Clay_AFFiNEAI_SDXL1_CLAYMATION.safetensors',
],
},
},
],
},
{
name: 'debug:action:fal-sdturbo-pixel',
action: 'image',
model: 'fast-turbo-diffusion',
messages: [
{
role: 'user',
content: 'pixel art, very high detail, masterpiece, {{content}}',
params: {
lora: ['https://models.affine.pro/fal/pixel-art-xl-v1.1.safetensors'],
},
},
],
},
{
name: 'debug:action:fal-sdturbo-sketch',
action: 'image',
model: 'fast-turbo-diffusion',
messages: [
{
role: 'user',
content: 'sketch for art examination, {{content}}',
params: {
lora: [
'https://models.affine.pro/fal/sketch_for_art_examination.safetensors',
],
},
},
],
},
{
name: 'debug:action:fal-sdturbo-fantasy',
action: 'image',
model: 'fast-turbo-diffusion',
messages: [
{
role: 'user',
content: 'fansty world, {{content}}',
params: {
lora: [
'https://models.affine.pro/fal/fansty%20world-000020.safetensors',
],
},
},
],
},
{
name: 'Summary',
action: 'Summary',