From 48e11cf843928f6e6e5c73d62ff02f9130308930 Mon Sep 17 00:00:00 2001 From: Jaret Burkett Date: Wed, 21 May 2025 03:10:33 -0600 Subject: [PATCH] Fallback unwrapping logic if fails --- toolkit/accelerator.py | 9 ++++++--- 1 file changed, 6 insertions(+), 3 deletions(-) diff --git a/toolkit/accelerator.py b/toolkit/accelerator.py index ebcf0095..0736f016 100644 --- a/toolkit/accelerator.py +++ b/toolkit/accelerator.py @@ -11,7 +11,10 @@ def get_accelerator() -> Accelerator: return global_accelerator def unwrap_model(model): - accelerator = get_accelerator() - model = accelerator.unwrap_model(model) - model = model._orig_mod if is_compiled_module(model) else model + try: + accelerator = get_accelerator() + model = accelerator.unwrap_model(model) + model = model._orig_mod if is_compiled_module(model) else model + except Exception as e: + pass return model