Commit 1a23dc32 authored by AUTOMATIC's avatar AUTOMATIC
Browse files

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

parent 4fdacd31
Loading
Loading
Loading
Loading
+1 −0
Original line number Diff line number Diff line
@@ -333,6 +333,7 @@ def load_model(checkpoint_info=None):

    timer = Timer()

    sd_model = None
    try:
        with sd_disable_initialization.DisableInitialization():
            sd_model = instantiate_from_config(sd_config.model)