Update app.py

app.py

@@ -100,25 +100,24 @@ class ModelConverter:
     def convert_model(self, input_model_id: str) -> Tuple[bool, Optional[str]]:
         """
         Convert the model to ONNX format, always exporting attention maps.
-        Also relocate any generation parameters into generation_config.json
-        and suppress JIT tracer warnings.
+        Relocate generation parameters and suppress tracer warnings.
+        Strip out the static relocation warning from stderr.
         """
         try:
-            # 1. Clone or prepare a local copy of the model to adjust configs
+            # 1. Prepare a local copy to adjust configs
             model_dir = self.config.repo_path / "models" / input_model_id
             model_dir.mkdir(parents=True, exist_ok=True)
 
             # 2. Load and relocate generation parameters
             config = AutoConfig.from_pretrained(input_model_id)
             gen_cfg = GenerationConfig.from_model_config(config)
-            # Remove generation-specific keys from model config
             for key in gen_cfg.to_dict().keys():
                 if hasattr(config, key):
                     setattr(config, key, None)
             config.save_pretrained(model_dir)
             gen_cfg.save_pretrained(model_dir)
 
-            # 3. Build the conversion command
+            # 3. Build and run the conversion command
             cmd = [
                 sys.executable,
                 "-m", "scripts.convert",
@@ -127,8 +126,6 @@ class ModelConverter:
                 "--model_id", input_model_id,
                 "--output_attentions",
             ]
-
-            # 4. Run the conversion
             result = subprocess.run(
                 cmd,
                 cwd=self.config.repo_path,
@@ -137,10 +134,18 @@ 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)
+
             if result.returncode != 0:
-                return False, result.stderr
+                return False, stderr
 
-            return True, result.stderr
+            return True, stderr
 
         except Exception as e:
             return False, str(e)
@@ -250,3 +255,4 @@ def main():
 
 if __name__ == "__main__":
     main()
+