bobmaster/phpbb_ailabs
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

Make LocalAI Stable Diffusion work in plugin.

Browse source
This commit is contained in:
hibobmaster 2023-12-24 22:44:02 +08:00
parent 625c47ab44
commit 49a3185585
Signed by: bobmaster
SSH key fingerprint: SHA256:5ZYgd8fg+PcNZNy4SzcSKu5JtqZyBF8kUhY7/k2viDk
1 changed files with 6 additions and 24 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

30
privet/ailabs/controller/dalle.php
View file

@ -66,22 +66,10 @@ class dalle extends GenericController
protected function prepare($opts) protected function prepare($opts)
{ {
if (filter_var($this->job['request'], FILTER_VALIDATE_URL)) {
// https://platform.openai.com/docs/api-reference/images/create-variation $opts += [
// The image to use as the basis for the variation(s). Must be a valid PNG file, less than 4MB, and square. 'prompt' => trim($this->job['request']),
$image = curl_file_create($this->job['request'], 'image/png'); ];
$opts += [
'image' => $image,
'n' => $this->cfg->n,
'response_format' => $this->cfg->response_format,
];
} else {
$opts += [
'prompt' => trim($this->job['request']),
'n' => $this->cfg->n,
'response_format' => $this->cfg->response_format,
];
}
return $opts; return $opts;
} }
@ -93,13 +81,7 @@ class dalle extends GenericController
$result = new resultSubmit(); $result = new resultSubmit();
if (empty($opts['image'])) { $result->response = $api->sendRequest($this->cfg->url_generations, 'POST', $opts);
// https://api.openai.com/v1/images/generations
$result->response = $api->sendRequest($this->cfg->url_generations, 'POST', $opts);
} else {
// https://api.openai.com/v1/images/variations
$result->response = $api->sendRequest($this->cfg->url_variations, 'POST', $opts);
}
$result->responseCodes = $api->responseCodes; $result->responseCodes = $api->responseCodes;
@ -173,4 +155,4 @@ class dalle extends GenericController
return $result; return $result;
} }
} }