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 pull request #175 from bakkot/unicode

Browse Source 
read/write plain text files in utf-8, not ascii
This commit is contained in:
Lincoln Stein 2022-08-29 07:03:55 -04:00 committed by GitHub
commit fd13526454
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
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

@ -90,7 +90,7 @@ class DreamServer(BaseHTTPRequestHandler):
post_data['initimg'] = '' # Don't send init image back post_data['initimg'] = '' # Don't send init image back
# Append post_data to log # 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: for output in outputs:
log.write(f"{output[0]}: {json.dumps(post_data)}\n") log.write(f"{output[0]}: {json.dumps(post_data)}\n")

4
scripts/dream.py
View File

@ -69,7 +69,7 @@ def main():
if opt.infile: if opt.infile:
try: try:
if os.path.isfile(opt.infile): if os.path.isfile(opt.infile):
infile = open(opt.infile, 'r') infile = open(opt.infile, 'r', encoding='utf-8')
elif opt.infile == '-': # stdin elif opt.infile == '-': # stdin
infile = sys.stdin infile = sys.stdin
else: else:
@ -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] log_lines = [f'{r[0]}: {prompt} -S{r[1]}\n' for r in results]
print(*log_lines, sep='') 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) file.writelines(log_lines)