Commit f5bcaf6a authored by gd1551's avatar gd1551 Committed by GitHub

import pathlib

parent d1f64ed9
...@@ -13,6 +13,7 @@ from sentry_sdk.integrations.threading import ThreadingIntegration ...@@ -13,6 +13,7 @@ from sentry_sdk.integrations.threading import ThreadingIntegration
from hydra_node.models import StableDiffusionModel, DalleMiniModel from hydra_node.models import StableDiffusionModel, DalleMiniModel
import traceback import traceback
import zlib import zlib
from pathlib import Path
model_map = {"stable-diffusion": StableDiffusionModel, "dalle-mini": DalleMiniModel} model_map = {"stable-diffusion": StableDiffusionModel, "dalle-mini": DalleMiniModel}
......
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