Mirrors
/
waifu-diffusion
mirror of https://github.com/harubaru/waifu-diffusion.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Cleanup test code.

This commit is contained in:
Maw-Fox 2022-11-11 17:17:50 -07:00
parent 120d406355
commit 6480336d2c
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
trainer/diffusers_trainer.py
View File

@ -299,9 +299,9 @@ class ImageStore:
# gets caption by removing the extension from the filename and replacing it with .txt
def get_caption(self, ref: Tuple[int, int, int]) -> str:
#filename = re.sub('\.[^/.]+$', '', self.image_files[ref[0]]) + '.txt'
#with open(filename, 'r', encoding='UTF-8') as f:
return ''
filename = re.sub('\.[^/.]+$', '', self.image_files[ref[0]]) + '.txt'
with open(filename, 'r', encoding='UTF-8') as f:
return f.read()
# ====================================== #