style: fix

Haidra-Org · Jun 10, 2024 · ec63577 · ec63577
1 parent 22135a3
commit ec63577
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 5 deletions.
diff --git a/horde/classes/kobold/genstats.py b/horde/classes/kobold/genstats.py
@@ -1,6 +1,6 @@
-from datetime import datetime, timedelta
+from datetime import datetime
 
-from sqlalchemy import Enum, func
+from sqlalchemy import Enum
 
 from horde.enums import ImageGenState
 from horde.flask import db

diff --git a/horde/classes/stable/genstats.py b/horde/classes/stable/genstats.py
@@ -1,10 +1,9 @@
-from datetime import datetime, timedelta
+from datetime import datetime
 
-from sqlalchemy import Enum, func
+from sqlalchemy import Enum
 
 from horde.enums import ImageGenState
 from horde.flask import db
-from horde.model_reference import model_reference
 
 
 class ImageGenerationStatisticPP(db.Model):

diff --git a/horde/model_reference.py b/horde/model_reference.py
@@ -55,9 +55,11 @@ def call_function(self):
                             self.nsfw_models.add(model)
                         if self.reference[model].get("type") == "controlnet":
                             self.controlnet_models.add(model)
+
                 break
             except Exception as e:
                 logger.error(f"Error when downloading nataili models list: {e}")
+
         for _riter in range(10):
             try:
                 self.text_reference = requests.get(