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

simplet2i: changed image file handling to work as stated in the [docs](https://pillow.readthedocs.io/en/stable/reference/open_files.html)

Browse Source
This commit is contained in:
yun saki 2022-08-26 14:13:16 +02:00
parent 89805a5239
commit 5129f256a3
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
ldm/simplet2i.py
View File

@ -545,7 +545,9 @@ class T2I:
return model return model
def _load_img(self, path): def _load_img(self, path):
image = Image.open(path).convert('RGB') with Image.open(path) as img:
image = img.convert("RGB")
w, h = image.size w, h = image.size
print(f'loaded input image of size ({w}, {h}) from {path}') print(f'loaded input image of size ({w}, {h}) from {path}')
w, h = map( w, h = map(