From e1c131aaf3c7bd4f17fafe9c62eb74cb41d1bf2c Mon Sep 17 00:00:00 2001 From: Lutz Roeder Date: Fri, 9 Dec 2022 18:20:13 -0800 Subject: [PATCH] Add TorchScript test file (#842) --- test/models.json | 29 +++++++++++++++++++++++++++++ 1 file changed, 29 insertions(+) diff --git a/test/models.json b/test/models.json index 76786b442b..704d79e5d2 100644 --- 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",