rixty/SillyTavern
1
0
Fork 0
mirror of https://github.com/SillyTavern/SillyTavern.git synced 2025-06-05 21:59:27 +02:00
Code Issues Packages Projects Releases Wiki Activity

/sd face now gens 2:3 ratio with width as base.

Browse Source
This commit is contained in:
RossAscends
2023-05-17 19:50:41 +09:00
parent 5040a0c66e
commit 2254bb2c7d
1 changed files with 8 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
public/scripts/extensions/stable-diffusion/index.js
View File

@ -367,6 +367,11 @@ async function generatePicture(_, trigger, message, callback) {
const quiet_prompt = getQuietPrompt(generationMode, trigger); const quiet_prompt = getQuietPrompt(generationMode, trigger);
const context = getContext(); const context = getContext();
if (trigger === 'face') {
var prevSDHeight = extension_settings.sd.height;
extension_settings.sd.height = extension_settings.sd.width * 1.5;
}
try { try {
const prompt = trigger == 'raw_last' ? message || getRawLastMessage(context) : processReply(await new Promise( const prompt = trigger == 'raw_last' ? message || getRawLastMessage(context) : processReply(await new Promise(
async function promptPromise(resolve, reject) { async function promptPromise(resolve, reject) {
@ -389,6 +394,7 @@ async function generatePicture(_, trigger, message, callback) {
throw new Error('SD prompt text generation failed.') throw new Error('SD prompt text generation failed.')
} }
finally { finally {
extension_settings.sd.height = prevSDHeight;
context.activateSendButtons(); context.activateSendButtons();
showSwipeButtons(); showSwipeButtons();
} }
@ -403,6 +409,7 @@ async function sendGenerationRequest(prompt, callback) {
} }
async function generateExtrasImage(prompt, callback) { async function generateExtrasImage(prompt, callback) {
console.log(extension_settings.sd);
const url = new URL(getApiUrl()); const url = new URL(getApiUrl());
url.pathname = '/api/image'; url.pathname = '/api/image';
const result = await fetch(url, { const result = await fetch(url, {
@ -595,6 +602,7 @@ $("#sd_dropdown [id]").on("click", function () {
else if (id == "sd_face") { else if (id == "sd_face") {
console.log("doing /sd face"); console.log("doing /sd face");
generatePicture('sd', 'face'); generatePicture('sd', 'face');
} }
else if (id == "sd_me") { else if (id == "sd_me") {