Commit 1a23dc32 authored by AUTOMATIC's avatar AUTOMATIC

possible fix for fallback for fast model creation from config, attempt 2

parent 4fdacd31
...@@ -333,6 +333,7 @@ def load_model(checkpoint_info=None): ...@@ -333,6 +333,7 @@ def load_model(checkpoint_info=None):
timer = Timer() timer = Timer()
sd_model = None
try: try:
with sd_disable_initialization.DisableInitialization(): with sd_disable_initialization.DisableInitialization():
sd_model = instantiate_from_config(sd_config.model) sd_model = instantiate_from_config(sd_config.model)
......
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