From 8cfa85a06ba72c92c3719b72aa94b9dcf545d0e0 Mon Sep 17 00:00:00 2001 From: Giuseppe Franco Date: Mon, 20 Nov 2023 16:19:50 +0000 Subject: [PATCH] Fix (export):remove unused attribute in debug flow --- src/brevitas/export/onnx/debug.py | 2 -- 1 file changed, 2 deletions(-) diff --git a/src/brevitas/export/onnx/debug.py b/src/brevitas/export/onnx/debug.py index 85acba31b..9f669c43c 100644 --- a/src/brevitas/export/onnx/debug.py +++ b/src/brevitas/export/onnx/debug.py @@ -51,7 +51,5 @@ def enable_debug(model, input_enabled=True, output_enabled=True, proxy_level=Fal for name, module in model.named_modules(): if hasattr(module, "export_debug_name") and filter_fn(module): module.export_debug_name = name - module.export_input_debug = input_enabled - module.export_output_debug = output_enabled module.register_forward_hook(hook) return hook