Skip to content

H
Hydra Node Http
Commit 96ac3de2 authored by kurumuz's avatar kurumuz
Browse files
Options

do the thing

parent a6b7f0ac
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 1 deletion
hydra_node/config.py
View file @ 96ac3de2
...@@ -10,7 +10,7 @@ from dotmap import DotMap ...@@ -10,7 +10,7 @@ from dotmap import DotMap
from icecream import ic from icecream import ic
from sentry_sdk import capture_exception from sentry_sdk import capture_exception
from sentry_sdk.integrations.threading import ThreadingIntegration from sentry_sdk.integrations.threading import ThreadingIntegration
from hydra_node.models import StableDiffusionModel, DalleMiniModel, BasedformerModel from hydra_node.models import StableDiffusionModel, DalleMiniModel, BasedformerModel, EmbedderModel
import traceback import traceback
import zlib import zlib
from pathlib import Path from pathlib import Path
...@@ -20,6 +20,7 @@ model_map = { ...@@ -20,6 +20,7 @@ model_map = {
"stable-diffusion": StableDiffusionModel, "stable-diffusion": StableDiffusionModel,
"dalle-mini": DalleMiniModel, "dalle-mini": DalleMiniModel,
"basedformer": BasedformerModel, "basedformer": BasedformerModel,
"embedder": EmbedderModel,
} }
def no_init(loading_code): def no_init(loading_code):
......
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