formatting

Signed-off-by: Ye Yu <yeyu@nvidia.com>

Author: yeyu-nvidia

modelopt/torch/export/plugins/hf_spec_export.py

@@ -92,7 +92,12 @@ def __init__(self, model: nn.Module):
         self.model = model
 
     @abstractmethod
-    def export(self, export_dir: Path | str, dtype: torch.dtype | None = None):
+    def export(
+        self,
+        export_dir: Path | str,
+        dtype: torch.dtype | None = None,
+        offline_specdec_input: dict | None = None,
+    ):
         """Export the model to the deployment format."""
         raise NotImplementedError("Subclasses must implement this method.")