Skip to content
Projects
Groups
Snippets
Help
Loading...
Help
Support
Keyboard shortcuts
?
Submit feedback
Sign in / Register
Toggle navigation
H
Hydra Node Http
Project overview
Project overview
Details
Activity
Releases
Repository
Repository
Files
Commits
Branches
Tags
Contributors
Graph
Compare
Locked Files
Issues
0
Issues
0
List
Boards
Labels
Service Desk
Milestones
Merge Requests
0
Merge Requests
0
CI / CD
CI / CD
Pipelines
Jobs
Schedules
Security & Compliance
Security & Compliance
Dependency List
License Compliance
Packages
Packages
List
Container Registry
Analytics
Analytics
CI / CD
Code Review
Insights
Issues
Repository
Value Stream
Wiki
Wiki
Snippets
Snippets
Members
Members
Collapse sidebar
Close sidebar
Activity
Graph
Create a new issue
Jobs
Commits
Issue Boards
Open sidebar
novelai-storage
Hydra Node Http
Commits
c662337d
Commit
c662337d
authored
Aug 06, 2022
by
novelailab
Browse files
Options
Browse Files
Download
Email Patches
Plain Diff
tracebacks everywhere
parent
367c364d
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
2 additions
and
0 deletions
+2
-0
hydra_node/config.py
hydra_node/config.py
+2
-0
No files found.
hydra_node/config.py
View file @
c662337d
...
@@ -11,6 +11,7 @@ from icecream import ic
...
@@ -11,6 +11,7 @@ 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
from
hydra_node.models
import
StableDiffusionModel
,
DalleMiniModel
import
traceback
model_map
=
{
"stable_diffusion"
:
StableDiffusionModel
,
"dalle-mini"
:
DalleMiniModel
}
model_map
=
{
"stable_diffusion"
:
StableDiffusionModel
,
"dalle-mini"
:
DalleMiniModel
}
...
@@ -105,6 +106,7 @@ def init_config_model():
...
@@ -105,6 +106,7 @@ def init_config_model():
model
=
model_map
[
config
.
model_name
](
config
)
model
=
model_map
[
config
.
model_name
](
config
)
except
Exception
as
e
:
except
Exception
as
e
:
traceback
.
print_exc
()
logger
.
error
(
f
"Failed to load model: {str(e)}"
)
logger
.
error
(
f
"Failed to load model: {str(e)}"
)
capture_exception
(
e
)
capture_exception
(
e
)
#exit gunicorn
#exit gunicorn
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment