Update app.py

app.py

@@ -187,7 +187,7 @@ class LLMService:
 
         model_id_or_path = "stablediffusionapi/realistic-vision-v51"
         self.vae_pipe = StableDiffusionImg2ImgPipeline.from_pretrained(model_id_or_path, torch_dtype=torch.float16)
-        self.vae_pipe = pipe.to(self.vit_sd_device)
+        self.vae_pipe = self.vae_pipe.to(self.vit_sd_device)
 
         self.boi_token_id = self.tokenizer.encode(BOI_TOKEN, add_special_tokens=False)[0]
         self.eoi_token_id = self.tokenizer.encode(EOI_TOKEN, add_special_tokens=False)[0]