Commit c662337d authored by novelailab's avatar novelailab

tracebacks everywhere

parent 367c364d
......@@ -11,6 +11,7 @@ from icecream import ic
from sentry_sdk import capture_exception
from sentry_sdk.integrations.threading import ThreadingIntegration
from hydra_node.models import StableDiffusionModel, DalleMiniModel
import traceback
model_map = {"stable_diffusion": StableDiffusionModel, "dalle-mini": DalleMiniModel}
......@@ -105,6 +106,7 @@ def init_config_model():
model = model_map[config.model_name](config)
except Exception as e:
traceback.print_exc()
logger.error(f"Failed to load model: {str(e)}")
capture_exception(e)
#exit gunicorn
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment