Skip to content

Commit d0d95d3

Browse files
committed
make initimg appear in web log
1 parent b90a215 commit d0d95d3

File tree

2 files changed

+4
-2
lines changed

2 files changed

+4
-2
lines changed

ldm/dream/server.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -87,7 +87,7 @@ def do_POST(self):
8787
# across images generated by each call to prompt2img(), so we define it in
8888
# the outer scope of image_done()
8989
config = post_data.copy() # Shallow copy
90-
config['initimg'] = ''
90+
config['initimg'] = config.pop('initimg_name','')
9191

9292
images_generated = 0 # helps keep track of when upscaling is started
9393
images_upscaled = 0 # helps keep track of when upscaling is completed

static/dream_web/index.js

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -19,7 +19,8 @@ function appendOutput(src, seed, config) {
1919
outputNode.addEventListener('click', () => {
2020
let form = document.querySelector("#generate-form");
2121
for (const [k, v] of new FormData(form)) {
22-
form.querySelector(`*[name=${k}]`).value = config[k];
22+
if (k == 'initimg') { continue; }
23+
form.querySelector(`*[name=${k}]`).value = config[k];
2324
}
2425
document.querySelector("#seed").value = seed;
2526

@@ -59,6 +60,7 @@ async function generateSubmit(form) {
5960

6061
// Convert file data to base64
6162
let formData = Object.fromEntries(new FormData(form));
63+
formData.initimg_name = formData.initimg.name
6264
formData.initimg = formData.initimg.name !== '' ? await toBase64(formData.initimg) : null;
6365

6466
let strength = formData.strength;

0 commit comments

Comments
 (0)