Sync with tops unit test for stable_diffusion.

DeepLink-org · Jan 17, 2024 · ba82f13 · ba82f13
1 parent 0bcf7fb
commit ba82f13
Show file tree

Hide file tree

Showing 2 changed files with 0 additions and 3 deletions.
diff --git a/dicp/test/model/stable_diffusion/ascendgraph_output.pt b/dicp/test/model/stable_diffusion/ascendgraph_output.pt
diff --git a/dicp/test/model/test_stable_diffusion.py b/dicp/test/model/test_stable_diffusion.py
@@ -43,9 +43,6 @@ def new_torch_variable_python_type(self):
         dicp_pipe = StableDiffusionPipeline.from_pretrained(model_path).to(device)
         dicp_pipe.text_encoder = torch.compile(dicp_pipe.text_encoder, backend=backend)
         dicp_pipe.unet = torch.compile(dicp_pipe.unet, backend=backend)
-        if backend == "ascendgraph":
-            dicp_pipe.vae.decoder = torch.compile(dicp_pipe.vae.decoder, backend=backend)
-
         dicp_image = dicp_pipe(prompt, num_inference_steps=num_inference_steps).images[0]
         if backend == "ascendgraph":
             standard_output = torch.load("stable_diffusion/ascendgraph_output.pt")