Commit 96ac3de2 authored by kurumuz's avatar kurumuz

do the thing

parent a6b7f0ac
......@@ -10,7 +10,7 @@ from dotmap import DotMap
from icecream import ic
from sentry_sdk import capture_exception
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 zlib
from pathlib import Path
......@@ -20,6 +20,7 @@ model_map = {
"stable-diffusion": StableDiffusionModel,
"dalle-mini": DalleMiniModel,
"basedformer": BasedformerModel,
"embedder": EmbedderModel,
}
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