From bc58807b14555df212041ed0dc38e7b3bd5c7acf Mon Sep 17 00:00:00 2001 From: Zach Mueller Date: Fri, 27 Oct 2023 16:52:03 +0000 Subject: [PATCH] Remove link --- src/accelerate/accelerator.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/accelerate/accelerator.py b/src/accelerate/accelerator.py index cc34cd5cb1c..5bb96d8c224 100755 --- a/src/accelerate/accelerator.py +++ b/src/accelerate/accelerator.py @@ -1372,7 +1372,7 @@ def prepare_model(self, model: torch.nn.Module, device_placement: bool = None, e with torch.no_grad(): convert_model(model) model._converted_to_transformer_engine = True - model._original_forward = model.forward + #model._original_forward = model.forward kwargs = self.fp8_recipe_handler.to_kwargs() if self.fp8_recipe_handler is not None else {} if "fp8_format" in kwargs: