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

bugfix: look in the correct directory when searching for wildcard

pull/2/head
AUTOMATIC 2022-10-22 15:10:04 +03:00
parent e541bf5af8
commit 9e165c8c91
1 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
scripts/wildcards.py
View File

@ -5,6 +5,8 @@ import sys
from modules import scripts
warned_about_files = {}
wildcard_dir = scripts.basedir()
class WildcardsScript(scripts.Script):
def title(self):
@ -14,10 +16,10 @@ class WildcardsScript(scripts.Script):
return scripts.AlwaysVisible
def replace_wildcard(self, text):
if " " in text or len(text)==0:
if " " in text or len(text) == 0:
return text
replacement_file = os.path.join(scripts.basedir(), f"wildcards/{text}.txt")
replacement_file = os.path.join(wildcard_dir, "wildcards", f"{text}.txt")
if os.path.exists(replacement_file):
with open(replacement_file, encoding="utf8") as f:
return random.choice(f.read().splitlines())