Commit 4fdacd31 authored by AUTOMATIC's avatar AUTOMATIC
Browse files

possible fix for fallback for fast model creation from config

parent 95409169

modules/sd_models.py

+3 −0
Original line number Diff line number Diff line
@@ -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)

CRA Git | Maintained and supported by SUSTech CRA and CCSE