This commit is contained in:
hyper 2024-10-25 16:32:07 +08:00
parent 9cc2dfd754
commit 4431bab82e
1 changed files with 2 additions and 1 deletions

View File

@ -44,12 +44,13 @@ export const defaultProdiaParams: ProdiaParams = {
export async function fetchProdia( export async function fetchProdia(
params: Partial<ProdiaParams>, params: Partial<ProdiaParams>,
endpoint = 'sdxl/generate'
) { ) {
let endpoint = 'sdxl/generate';
if(params.input_image){ if(params.input_image){
const data = await fs.readFile(params.input_image); const data = await fs.readFile(params.input_image);
params.imageData = data.toString('base64'); params.imageData = data.toString('base64');
delete params.input_image; delete params.input_image;
endpoint = 'sdxl/transform';
} }
const resp = await fetch(`https://api.prodia.com/v1/${endpoint}`, { const resp = await fetch(`https://api.prodia.com/v1/${endpoint}`, {
method: 'POST', method: 'POST',