File size: 2,519 Bytes
4905b6b 660842c 4905b6b 660842c 4905b6b 660842c 4905b6b 660842c 4905b6b 660842c |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 |
import { StableDiffusionParams } from "@/types"
import { CharacterExpression, CharacterGender, CharacterPose, GenerationMode, HairColor, HairStyle } from './config'
import { generateSeed } from "./generateSeed"
export function getStableDiffusionParams({
generationMode,
negativePrompt,
characterAge,
characterExpression,
characterGender,
characterHairColor,
characterHairStyle,
characterName,
characterPose
}: {
generationMode: GenerationMode
negativePrompt: string
characterAge: number
characterExpression: CharacterExpression
characterGender: CharacterGender
characterHairColor: HairColor
characterHairStyle: HairStyle
characterName: string
characterPose: CharacterPose
}): StableDiffusionParams {
const baseParams: StableDiffusionParams = {
prompt: "",
negativePrompt: "",
guidanceScale: 9,
seed: generateSeed(),
width: 1024,
height: 1024,
numInferenceSteps: 20,
hfApiKey: "",
}
const params: StableDiffusionParams = {
...baseParams,
...generationMode === "characters" ? {
prompt: [
`${characterAge} years old $${characterGender}`,
`${characterHairColor} ${characterHairStyle}`,
`looking ${characterExpression}`,
`${characterPose}`,
`named ${characterName}`,
'beautiful',
'award winning',
'sharp',
'crisp',
'centered',
'aligned'
].filter(x => x).join(", "),
negativePrompt: [
negativePrompt,
'drawing',
'painting',
'unrealistic',
'sitting',
'chair',
'3D render',
'unaligned',
'cropped',
'bad hands',
'wrong hands',
'deformed',
'glitch',
'blurry',
'overexposed'
].join(", "),
} : {
prompt: [
'picture of a single',
prompt,
'3D render',
'logo',
'ios icon',
'illustration',
'vector graphics',
'svg',
'beautiful',
'award winning',
'sharp',
'crisp',
'centered',
'aligned',
].join(", "),
negativePrompt: [
negativePrompt,
'photo',
'gradient',
'many',
'realistic',
'shadow',
'multiple',
'various',
'unaligned',
'cropped',
'bad hands',
'wrong hands',
'deformed',
'glitch',
'blurry',
'overexposed'
].join(", "),
}
}
return params
} |