From 0eb4f5b86dca37a1fc9230244c14e8cf65821854 Mon Sep 17 00:00:00 2001 From: Qing Date: Mon, 5 Feb 2024 21:29:40 +0800 Subject: [PATCH] fix revision==fp16 fallback to main --- iopaint/model/utils.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/iopaint/model/utils.py b/iopaint/model/utils.py index 95c1184..73465e8 100644 --- a/iopaint/model/utils.py +++ b/iopaint/model/utils.py @@ -983,7 +983,11 @@ def handle_from_pretrained_exceptions(func, **kwargs): except ValueError as e: if "You are trying to load the model files of the `variant=fp16`" in str(e): logger.info("variant=fp16 not found, try revision=fp16") - return func(**{**kwargs, "variant": None, "revision": "fp16"}) + try: + return func(**{**kwargs, "variant": None, "revision": "fp16"}) + except Exception as e: + logger.info("revision=fp16 not found, try revision=main") + return func(**{**kwargs, "variant": None, "revision": "main"}) raise e except OSError as e: previous_traceback = traceback.format_exc()