1580158
/
stable-diffusion-webui-inspiration
mirror of https://github.com/yfszzx/stable-diffusion-webui-inspiration.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Ignore headline of artist.csv from automatic1111 repository

pull/20/head
ir31k0 2022-11-15 22:37:02 +01:00
parent 5139284570
commit 4e704b5220
1 changed files with 9 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
scripts/inspiration.py
View File

@ -39,10 +39,15 @@ class Script(scripts.Script):
os.makedirs(path)
f = open(file.name, "r", encoding='utf-8')
line = f.readline()
while len(line) > 0:
name = line.rstrip("\n").split(",")[0]
line = f.readline()
artist_path = os.path.join(path, name)
while len(line) > 0:
line = line.rstrip("\n")
# Ignore headline of artist.csv from automatic1111 repository
if line == "artist,score,category":
line = f.readline()
continue
name = line.split(",")[0]
line = f.readline()
artist_path = os.path.join(path, name)
if not os.path.exists(artist_path):
os.mkdir(artist_path)
if len(os.listdir(artist_path)) >= opts.inspiration_max_samples: