Commit d8c57f5b authored by novelailab's avatar novelailab

fix stable diffusion name

parent c662337d
...@@ -13,7 +13,7 @@ from sentry_sdk.integrations.threading import ThreadingIntegration ...@@ -13,7 +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
model_map = {"stable_diffusion": StableDiffusionModel, "dalle-mini": DalleMiniModel} model_map = {"stable-diffusion": StableDiffusionModel, "dalle-mini": DalleMiniModel}
def no_init(loading_code): def no_init(loading_code):
def dummy(self): def dummy(self):
......
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