Update src/pipeline.py

src/pipeline.py

@@ -113,9 +113,9 @@ def load_pipeline(pipeline=None) -> StableDiffusionXLPipeline:
         print("creating latent")
         latents = torch.randn(1, 4, 128, 128, device='cuda', requires_grad=False)
         print("decoding ")
-        res_sd = pipe.vae.decode(latents)[0]
+        res_sd = pipeline.vae.decode(latents)[0]
         print("compressing ")
-        res_taesd = compress(latents.mul(pipe.vae.config.scaling_factor)).mul(2).sub(1)
+        res_taesd = compress(latents.mul(pipeline.vae.config.scaling_factor)).mul(2).sub(1)
         print("checking loss")
         loss = loss_fn(res_taesd, res_sd.float())
         optimizer.zero_grad()