Update app.py

app.py

@@ -13,7 +13,7 @@ from huggingface_hub import HfApi, whoami
 from torch.jit import TracerWarning
 from transformers import AutoConfig, GenerationConfig
 
-# Suppress TorchScript tracer warnings globally
+# Suppress TorchScript tracer warnings in parent process
 warnings.filterwarnings("ignore", category=TracerWarning)
 
 logging.basicConfig(level=logging.INFO)
@@ -100,15 +100,15 @@ class ModelConverter:
     def convert_model(self, input_model_id: str) -> Tuple[bool, Optional[str]]:
         """
         Convert the model to ONNX format, always exporting attention maps.
-        Relocate generation parameters and suppress tracer warnings.
-        Strip out the static relocation warning from stderr.
+        Relocate generation parameters and suppress warnings in subprocess.
+        Strip out static relocation and tracer warnings from stderr.
         """
         try:
-            # 1. Prepare a local copy to adjust configs
+            # Prepare local model folder for config edits
             model_dir = self.config.repo_path / "models" / input_model_id
             model_dir.mkdir(parents=True, exist_ok=True)
 
-            # 2. Load and relocate generation parameters
+            # Reload and relocate generation parameters
             config = AutoConfig.from_pretrained(input_model_id)
             gen_cfg = GenerationConfig.from_model_config(config)
             for key in gen_cfg.to_dict().keys():
@@ -117,9 +117,10 @@ class ModelConverter:
             config.save_pretrained(model_dir)
             gen_cfg.save_pretrained(model_dir)
 
-            # 3. Build and run the conversion command
+            # Build command with warning suppression flag
             cmd = [
                 sys.executable,
+                "-W", "ignore::torch.jit.TracerWarning",
                 "-m", "scripts.convert",
                 "--quantize",
                 "--trust_remote_code",
@@ -134,13 +135,15 @@ class ModelConverter:
                 env=os.environ.copy(),
             )
 
-            # 4. Filter out the relocation warning from stderr
-            filtered_lines = [
-                ln
-                for ln in result.stderr.splitlines()
-                if not ln.startswith("Moving the following attributes")
-            ]
-            stderr = "\n".join(filtered_lines)
+            # Filter stderr lines
+            filtered = []
+            for ln in result.stderr.splitlines():
+                if ln.startswith("Moving the following attributes"):  # relocation warning
+                    continue
+                if "TracerWarning" in ln:  # any tracer warnings
+                    continue
+                filtered.append(ln)
+            stderr = "\n".join(filtered)
 
             if result.returncode != 0:
                 return False, stderr
@@ -254,5 +257,4 @@ def main():
 
 
 if __name__ == "__main__":
-    main()
-
+    main()