Mirrored_Repos/InvokeAI
1
0
Fork 0
mirror of https://github.com/invoke-ai/InvokeAI synced 2024-08-30 20:32:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

Merge branch 'main' of github.com:lstein/stable-diffusion into main

Browse Source
This commit is contained in:
Lincoln Stein 2022-08-28 19:03:45 -04:00
commit 373a2d9c32
2 changed files with 11 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
ldm/dream/server.py
View File

@ -14,8 +14,14 @@ class DreamServer(BaseHTTPRequestHandler):
self.end_headers() self.end_headers()
with open("./static/dream_web/index.html", "rb") as content: with open("./static/dream_web/index.html", "rb") as content:
self.wfile.write(content.read()) self.wfile.write(content.read())
elif os.path.exists("." + self.path): else:
mime_type = mimetypes.guess_type(self.path)[0] path = "." + self.path
cwd = os.getcwd()
is_in_cwd = os.path.commonprefix((os.path.realpath(path), cwd)) == cwd
if not (is_in_cwd and os.path.exists(path)):
self.send_response(404)
return
mime_type = mimetypes.guess_type(path)[0]
if mime_type is not None: if mime_type is not None:
self.send_response(200) self.send_response(200)
self.send_header("Content-type", mime_type) self.send_header("Content-type", mime_type)
@ -24,8 +30,6 @@ class DreamServer(BaseHTTPRequestHandler):
self.wfile.write(content.read()) self.wfile.write(content.read())
else: else:
self.send_response(404) self.send_response(404)
else:
self.send_response(404)
def do_POST(self): def do_POST(self):
self.send_response(200) self.send_response(200)

6
static/dream_web/index.html
View File

@ -29,7 +29,7 @@
<option value="64">64</option> <option value="128">128</option> <option value="64">64</option> <option value="128">128</option>
<option value="192">192</option> <option value="256">256</option> <option value="192">192</option> <option value="256">256</option>
<option value="320">320</option> <option value="384">384</option> <option value="320">320</option> <option value="384">384</option>
<option value="448">448</option> <option value="512">512</option> <option value="448">448</option> <option value="512" selected>512</option>
<option value="576">576</option> <option value="640">640</option> <option value="576">576</option> <option value="640">640</option>
<option value="704">704</option> <option value="768">768</option> <option value="704">704</option> <option value="768">768</option>
<option value="832">832</option> <option value="896">896</option> <option value="832">832</option> <option value="896">896</option>
@ -40,7 +40,7 @@
<option value="64">64</option> <option value="128">128</option> <option value="64">64</option> <option value="128">128</option>
<option value="192">192</option> <option value="256">256</option> <option value="192">192</option> <option value="256">256</option>
<option value="320">320</option> <option value="384">384</option> <option value="320">320</option> <option value="384">384</option>
<option value="448">448</option> <option value="512">512</option> <option value="448">448</option> <option value="512" selected>512</option>
<option value="576">576</option> <option value="640">640</option> <option value="576">576</option> <option value="640">640</option>
<option value="704">704</option> <option value="768">768</option> <option value="704">704</option> <option value="768">768</option>
<option value="832">832</option> <option value="896">896</option> <option value="832">832</option> <option value="896">896</option>
@ -53,7 +53,7 @@
<input value="-1" type="number" id="seed" name="seed"> <input value="-1" type="number" id="seed" name="seed">
<button type="button" id="reset">&olarr;</button> <button type="button" id="reset">&olarr;</button>
<br> <br>
<label title="Strenght of the gfpgan algorithm ex: '1', --gfpgan startup flag is required." for="gfpgan_strength">GPFGAN Strength:</label> <label title="Strength of the gfpgan algorithm ex: '1', --gfpgan startup flag is required." for="gfpgan_strength">GPFGAN Strength:</label>
<input value="0.75" min="0" max="1" type="number" id="gfpgan_strength" name="gfpgan_strength" step="0.01"> <input value="0.75" min="0" max="1" type="number" id="gfpgan_strength" name="gfpgan_strength" step="0.01">
</fieldset> </fieldset>
</form> </form>