Update app.py

app.py

@@ -156,7 +156,7 @@ def generate(segment, image, prompt, size, guidance_scale, num_inference_steps,
         prompt_attention_mask = state["prompt_attention_mask"].to("cuda", dtype=torch.bfloat16)
         image_latents = state["image_latents"].to("cuda", dtype=torch.bfloat16)
     if segment==9:
-        pipe.trasformer.to('cpu')
+        pipe.transformer.to('cpu')
         torch.cuda.empty_cache()
         pipe.vae.to("cuda")
         latents = latents.to(pipe.vae.dtype) / pipe.vae.config.scaling_factor