import pathlib

f5bcaf6a · gd1551 · GitHub · d1f64ed9 · f5bcaf6a
Commit f5bcaf6a authored Aug 21, 2022 by gd1551 Committed by GitHub Aug 21, 2022
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 0 deletions

hydra_node/config.py hydra_node/config.py +1 -0

No files found.
--- a/hydra_node/config.py
+++ b/hydra_node/config.py
@@ -13,6 +13,7 @@ from sentry_sdk.integrations.threading import ThreadingIntegration
 from hydra_node.models import StableDiffusionModel, DalleMiniModel
 import traceback
 import zlib
+from pathlib import Path

 model_map = {"stable-diffusion": StableDiffusionModel, "dalle-mini": DalleMiniModel}