diff --git a/ldm/invoke/model_manager.py b/ldm/invoke/model_manager.py index 47b493f2955..92265bd7177 100644 --- a/ldm/invoke/model_manager.py +++ b/ldm/invoke/model_manager.py @@ -433,7 +433,7 @@ def _load_diffusers_model(self, mconfig): **fp_args, ) except OSError as e: - if str(e).startswith("fp16 is not a valid"): + if 'Revision Not Found' in str(e): pass else: print(