From 007d1980e7e9cf5905a5e436d7360e0b7f5e6648 Mon Sep 17 00:00:00 2001
From: Hitmare <hitmare@noreply@foxo.me>
Date: Sat, 11 Nov 2023 20:32:26 +0000
Subject: [PATCH] Updated sideload files

YAML files are now as well added into the list
---
 erisscan.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/erisscan.py b/erisscan.py
index 4342b90..34ff3be 100644
--- a/erisscan.py
+++ b/erisscan.py
@@ -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')