supporting embed config in extra_files

Reviewed By: sanjeevk42

Differential Revision: D28346869

fbshipit-source-id: b226acf5ee5d90be4ea183dc7de92133db4d5717

d2go/export/torchscript.py

@@ -73,7 +73,9 @@ def trace_and_save_torchscript(
             )
             torchscript_filename = mobile_optimization.torchscript_filename
             with _synced_local_file(torchscript_filename) as lite_path:
-                liteopt_model._save_for_lite_interpreter(lite_path)
+                liteopt_model._save_for_lite_interpreter(
+                    lite_path, _extra_files=_extra_files
+                )
             # liteopt_model(*inputs)  # sanity check
             op_names = torch.jit.export_opnames(liteopt_model)
             logger.info(
@@ -97,7 +99,7 @@ def trace_and_save_torchscript(
 
 
 class TorchscriptWrapper(nn.Module):
-    """"""
+    """ """
 
     def __init__(self, module, int8_backend=None):
         super().__init__()