Modify stable-diffusion's result format.

DeepLink-org · Jan 12, 2024 · 64cebe4 · 64cebe4
1 parent 2ac6c2e
commit 64cebe4
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 7 deletions.
diff --git a/dicp/test/model/stable_diffusion/topsgraph_output.pt b/dicp/test/model/stable_diffusion/topsgraph_output.pt
diff --git a/dicp/test/model/stable_diffusion/topsgraph_output.txt b/dicp/test/model/stable_diffusion/topsgraph_output.txt
diff --git a/dicp/test/model/test_stable_diffusion.py b/dicp/test/model/test_stable_diffusion.py
@@ -32,12 +32,10 @@ def test_inference(
         dicp_pipe.unet = torch.compile(dicp_pipe.unet, backend=backend)
         dicp_image = dicp_pipe(prompt, num_inference_steps=num_inference_steps).images[0]
         if backend == "ascendgraph":
-            with open("stable_diffusion/topsgraph_output.txt", "r") as f:
-                standard_output = eval(f.read())
+            standard_output = torch.load("stable_diffusion/ascendgraph_output.pt")
         elif backend == "topsgraph":
-            with open("stable_diffusion/topsgraph_output.txt", "r") as f:
-                standard_output = eval(f.read())
+            standard_output = torch.load("stable_diffusion/topsgraph_output.pt")
         else:
             raise ValueError("backend should in (ascendgrap, topsgraph)")
-        dicp_output = list(dicp_image.getdata())
-        assert dicp_output == standard_output
+        dicp_output = torch.tensor(list(dicp_image.getdata()))
+        assert torch.allclose(dicp_output, standard_output, equal_nan=True)