Add TorchScript test file (#842)

lutzroeder · Dec 10, 2022 · e1c131a · e1c131a
1 parent 703de61
commit e1c131a
Showing 1 changed file with 29 additions and 0 deletions.
diff --git a/test/models.json b/test/models.json
@@ -4932,6 +4932,35 @@
     "format":   "PyTorch v0.1.10",
     "link":     "https://github.com/MagicLeapResearch/SuperPointPretrainedNetwork"
   },
+  {
+    "type":     "pytorch",
+    "target":   "stable_diffusion_safety_checker.pt",
+    "source":   "https://github.com/lutzroeder/netron/files/10199037/stable_diffusion.zip[stable_diffusion_safety_checker.pt]",
+    "format":   "TorchScript v1.6",
+    "link":     "https://github.com/lutzroeder/netron/issues/842"
+  },
+  {
+    "type":     "pytorch",
+    "target":   "stable_diffusion_text_encoder.pt",
+    "source":   "https://github.com/lutzroeder/netron/files/10199037/stable_diffusion.zip[stable_diffusion_text_encoder.pt]",
+    "format":   "TorchScript v1.6",
+    "link":     "https://github.com/lutzroeder/netron/issues/842"
+  },
+  {
+    "type":     "pytorch",
+    "target":   "stable_diffusion_unet.pt",
+    "source":   "https://github.com/lutzroeder/netron/files/10199037/stable_diffusion.zip[stable_diffusion_unet.pt]",
+    "format":   "TorchScript v1.12",
+    "action":   "skip-render",
+    "link":     "https://github.com/lutzroeder/netron/issues/842"
+  },
+  {
+    "type":     "pytorch",
+    "target":   "stable_diffusion_vae_decoder.pt",
+    "source":   "https://github.com/lutzroeder/netron/files/10199037/stable_diffusion.zip[stable_diffusion_vae_decoder.pt]",
+    "format":   "TorchScript v1.6",
+    "link":     "https://github.com/lutzroeder/netron/issues/842"
+  },
   {
     "type":     "pytorch",
     "target":   "superpoint.pt",