Add TorchScript test file (#842)

lutzroeder · May 28, 2023 · f1c95a0 · f1c95a0
1 parent d110fb3
commit f1c95a0
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 0 deletions.
diff --git a/source/python.js b/source/python.js
@@ -4879,6 +4879,9 @@ python.Execution = class {
             }
             return [];
         });
+        this.registerFunction('torch.sqrt', function(x) {
+            return Math.sqrt(x);
+        });
         this.registerFunction('torch.slice', function(l, start, end, step) {
             if (!Array.isArray(l)) {
                 throw new python.Error('Slicing expected array');

diff --git a/test/models.json b/test/models.json
@@ -4392,6 +4392,13 @@
     "format":   "TorchScript v1.0",
     "link":     "https://github.com/jiexiong2016/GCNv2_SLAM"
   },
+  {
+    "type":     "pytorch",
+    "target":   "gpt2.pt",
+    "source":   "https://github.com/lutzroeder/netron/files/11585383/gpt2.pt.zip[gpt2.pt]",
+    "format":   "TorchScript v1.12",
+    "link":     "https://github.com/lutzroeder/netron/issues/842"
+  },
   {
     "type":     "pytorch",
     "target":   "inception_v3_traced.pt",