From ec63577bb59df7100031fbaf91d382093f798137 Mon Sep 17 00:00:00 2001 From: tazlin Date: Mon, 10 Jun 2024 16:51:11 -0400 Subject: [PATCH] style: fix --- horde/classes/kobold/genstats.py | 4 ++-- horde/classes/stable/genstats.py | 5 ++--- horde/model_reference.py | 2 ++ 3 files changed, 6 insertions(+), 5 deletions(-) diff --git a/horde/classes/kobold/genstats.py b/horde/classes/kobold/genstats.py index b0b9e28c..f6fc9b6e 100644 --- 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 index 1c964da5..8c217333 100644 --- 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 index a6459469..82e65a41 100644 --- 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(