sneedgroup-holder/openOutpaint
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

reuse random seeds

Browse source
This commit is contained in:
tim h 2022-12-13 19:30:40 -06:00
parent 2e151d2bbc
commit 9b23e3aa92
1 changed files with 24 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

25
js/ui/tool/dream.js
View file

@ -101,6 +101,8 @@ const _dream = async (endpoint, request) => {
} finally { } finally {
generating = false; generating = false;
} }
var responseSubdata = JSON.parse(data.info);
stableDiffusionData.lastSeeds.push(...responseSubdata.all_seeds);
return data.images; return data.images;
}; };
@ -256,6 +258,7 @@ const _generate = async (
}); });
imageCollection.inputElement.appendChild(interruptButton); imageCollection.inputElement.appendChild(interruptButton);
stableDiffusionData.lastSeeds = [];
images.push(...(await _dream(endpoint, requestCopy))); images.push(...(await _dream(endpoint, requestCopy)));
stopDrawingStatus = true; stopDrawingStatus = true;
at = 1; at = 1;
@ -276,6 +279,8 @@ const _generate = async (
if (at < 0) at = images.length - 1; if (at < 0) at = images.length - 1;
imageindextxt.textContent = `${at}/${images.length - 1}`; imageindextxt.textContent = `${at}/${images.length - 1}`;
var seed = stableDiffusionData.lastSeeds[at - 1];
seedbtn.title = "Use seed " + seed;
redraw(); redraw();
}; };
@ -284,6 +289,8 @@ const _generate = async (
if (at >= images.length) at = 0; if (at >= images.length) at = 0;
imageindextxt.textContent = `${at}/${images.length - 1}`; imageindextxt.textContent = `${at}/${images.length - 1}`;
var seed = stableDiffusionData.lastSeeds[at - 1];
seedbtn.title = "Use seed " + seed;
redraw(); redraw();
}; };
@ -455,7 +462,10 @@ const _generate = async (
// load the image data after defining the closure // load the image data after defining the closure
img.src = "data:image/png;base64," + images[at]; img.src = "data:image/png;base64," + images[at];
img.addEventListener("load", () => { img.addEventListener("load", () => {
const response = prompt("Enter new resource name", "Dream Resource"); const response = prompt(
"Enter new resource name",
"Dream Resource " + stableDiffusionData.lastSeeds[at - 1]
);
if (response) { if (response) {
tools.stamp.state.addResource(response, img); tools.stamp.state.addResource(response, img);
redraw(); // Redraw to avoid strange cursor behavior redraw(); // Redraw to avoid strange cursor behavior
@ -472,6 +482,14 @@ const _generate = async (
}); });
imageSelectMenu.appendChild(savebtn); imageSelectMenu.appendChild(savebtn);
const seedbtn = document.createElement("button");
seedbtn.textContent = "U";
seedbtn.title = "Use seed " + `${stableDiffusionData.lastSeeds[at - 1]}`;
seedbtn.addEventListener("click", () => {
sendSeed(at);
});
imageSelectMenu.appendChild(seedbtn);
nextQueue(initialQ); nextQueue(initialQ);
}; };
@ -1623,3 +1641,8 @@ window.onbeforeunload = async () => {
// Stop current generation on page close // Stop current generation on page close
if (generating) await fetch(`${host}${url}interrupt`, {method: "POST"}); if (generating) await fetch(`${host}${url}interrupt`, {method: "POST"});
}; };
function sendSeed(seedIndex) {
document.getElementById("seed").value =
stableDiffusionData.lastSeeds[seedIndex - 1];
}