Commit 4fdacd31 authored by AUTOMATIC's avatar AUTOMATIC

possible fix for fallback for fast model creation from config

parent 95409169
......@@ -337,6 +337,9 @@ def load_model(checkpoint_info=None):
with sd_disable_initialization.DisableInitialization():
sd_model = instantiate_from_config(sd_config.model)
except Exception as e:
pass
if sd_model is None:
print('Failed to create model quickly; will retry using slow method.', file=sys.stderr)
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