DiffusionAi/InvokeAI
1
0
Fork 0
mirror of https://github.com/invoke-ai/InvokeAI.git synced 2025-01-09 04:18:46 +08:00
Code Issues Packages Projects Releases Wiki Activity

write log files in utf-8, not ascii

Browse Source
This commit is contained in:
Kevin Gibbons 2022-08-28 19:57:13 -07:00
parent 9a8cd9684e
commit 7ae79b350e
2 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
ldm/dream/server.py
View File

@ -83,7 +83,7 @@ class DreamServer(BaseHTTPRequestHandler):
post_data['initimg'] = '' # Don't send init image back
# Append post_data to log
with open("./outputs/img-samples/dream_web_log.txt", "a") as log:
with open("./outputs/img-samples/dream_web_log.txt", "a", encoding="utf-8") as log:
for output in outputs:
log.write(f"{output[0]}: {json.dumps(post_data)}\n")

6
scripts/dream.py
View File

@ -69,7 +69,7 @@ def main():
if opt.infile:
try:
if os.path.isfile(opt.infile):
infile = open(opt.infile, 'r')
infile = open(opt.infile, 'r', encoding='utf-8')
elif opt.infile == '-': # stdin
infile = sys.stdin
else:
@ -217,7 +217,7 @@ def get_next_command(infile=None) -> str: #command string
command = command.strip()
print(f'#{command}')
return command
def dream_server_loop(t2i):
print('\n* --web was specified, starting web server...')
# Change working directory to the stable-diffusion directory
@ -244,7 +244,7 @@ def write_log_message(prompt, results, log_path):
log_lines = [f'{r[0]}: {prompt} -S{r[1]}\n' for r in results]
print(*log_lines, sep='')
with open(log_path, 'a') as file:
with open(log_path, 'a', encoding='utf-8') as file:
file.writelines(log_lines)