Updated sideload files

YAML files are now as well added into the list
This commit is contained in:
Hitmare 2023-11-11 20:32:26 +00:00
parent 7f2a6dbff1
commit 007d1980e7
1 changed files with 2 additions and 0 deletions

View File

@ -58,6 +58,8 @@ def scanfolder(path: str) -> list:
sideloaddict.append(f"{modelname}.preview.jpeg")
if os.path.exists(f"{modelfolder}{modelname}.preview.png"):
sideloaddict.append(f"{modelname}.preview.png")
if os.path.exists(f"{modelfolder}{modelname}.yaml"):
sideloaddict.append(f"{modelname}.yaml")
else:
print(f"Model {modelfile} found on CivitAi")
modelid = modelinfoarray.get('modelId')