raise error

huggingface · Jul 23, 2024 · 6d12238 · 6d12238
1 parent 5c27181
commit 6d12238
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/optimum/intel/ipex/modeling_base.py b/optimum/intel/ipex/modeling_base.py
@@ -632,9 +632,9 @@ def generate(self, *args, **kwargs):
             _patch_crop_past_key_values()
         try:
             result = super().generate(*args, **kwargs)
-        except Exception as error:
+        except Exception as e:
             _unpatch_crop_past_key_values()
-            assert False, f"IPEXModelForCausalLM generation failed due to {error}"
+            raise e
         _unpatch_crop_past_key_values()
         return result