Upload 2 files

app.py

@@ -196,20 +196,15 @@ with gr.Blocks(title="PTH to ONNX Converter") as demo:
     install()
     spandrel_extra_arches.install()
 
-    gr.Textbox(value="This tool is still in development. Outputs may be incorrect.", label="Notice")
-    # with gr.Accordion("Model Cheat Sheet", open=False):
-    #     gr.Markdown("""
-    #     After converting a model, be sure to click the log button above to check the output for any errors or warnings.
-    #     """)
-
     file_upload = gr.File(label="Upload a PyTorch model", file_types=['.pth', '.pt', '.safetensors'])
     metadata = gr.Textbox(value="Ready", label="File Information")
     
     dropdown_opset = gr.Dropdown(choices=[17, 18, 19, 20], value=20, label="Opset")
-    radio_fp16 = gr.Radio(choices=["True", "False"], value="False", label="FP16 - Export at half precision. Not supported by all models.")
-    radio_static = gr.Radio(choices=["True", "False"], value="False", label="Static Shapes - Some models may work better with static shapes. You may need to try both options.")
-    radio_slim = gr.Radio(choices=["True", "False"], value="False", label="OnnxSlim - Can potentially optimize the model further, but usually has little effect and can cause the model to not work correctly.")
+    radio_fp16 = gr.Radio(choices=["True", "False"], value="False", label="FP16 - Not supported by all models. Not very useful because FP16 TRT engines can still be built from FP32 ONNX models.")
+    radio_static = gr.Radio(choices=["True", "False"], value="False", label="Static Shapes - Might be required by some models, but can cause slower performance.")
+    radio_slim = gr.Radio(choices=["True", "False"], value="False", label="OnnxSlim - Can cause issues in some models. I have not yet found any cases where it helps. May remove in the future.")
     
+    gr.Markdown("After converting, click the logs button at the top to check for any errors or warnings.")
     process_button = gr.Button("Convert", interactive=True)
     file_output = gr.DownloadButton(label="💾 Download Converted Model", visible=False)
 

requirements.txt

@@ -1,6 +1,7 @@
 spandrel>=0.4.1
 spandrel-extra-arches>=0.2.0
 torch>=2.7.1
+numpy==2.2.6
 gradio
 onnx
 onnxruntime