Merge pull request #542 from ModECI/fix/pytorch_pin

Remove pytorch upper pin

Author: davidt0x

examples/MDF/abcd_torch.py

@@ -86,7 +86,7 @@ def forward(self, input: torch.Tensor):
 
 # Export the model
 fn = "ABCD_from_torch.onnx"
-torch_out = torch.onnx._export(
+torch.onnx.export(
     m_abcd,  # model being run
     input,  # model input (or a tuple for multiple inputs)
     fn,  # where to save the model (can be a file or file-like object)

setup.cfg

@@ -89,7 +89,7 @@ optional =
     Jinja2<3.1
     torchviz
     netron
-    torch<2.2.0,>=1.11.0
+    torch>=1.11.0
     torchvision
     h5py
 

src/modeci_mdf/interfaces/pytorch/importer.py

@@ -5,6 +5,7 @@
 
     Mike He, "From Models to Computation Graphs (Part I)", https://ad1024.space/articles/22
 """
+
 import inspect
 import logging
 
@@ -13,6 +14,7 @@
 
 import torch
 
+
 # We need to monkey patch the torch._C.Node class to add a __getitem__ method
 # This is for torch 2.0
 # From https://github.com/openai/CLIP/issues/79#issuecomment-1624202950
@@ -523,7 +525,16 @@ def pytorch_to_mdf(
     # Call out to a part of the ONNX exporter that simiplifies the graph before ONNX export.
     from torch.onnx.utils import _model_to_graph
     from torch.onnx import TrainingMode
-    from torch.onnx.symbolic_helper import _set_opset_version
+
+    # Seems they got rid of _set_opset_version in 2.2 or something, can't find a better way to do this
+    try:
+        from torch.onnx.symbolic_helper import _set_opset_version
+    except ImportError:
+
+        def _set_opset_version(version):
+            from torch.onnx._globals import GLOBALS
+
+            GLOBALS.export_onnx_opset_version = version
 
     try:
         from torch.onnx.symbolic_helper import _export_onnx_opset_version