placed the vase and refiner the same as i had on osiris

app.py

@@ -37,24 +37,33 @@ processor = AutoProcessor.from_pretrained("xtuner/llava-llama-3-8b-v1_1-transfor
 
 llava_model.generation_config.eos_token_id = 128009
 
-# Initialize Stable Diffusion pipelines
-base = DiffusionPipeline.from_pretrained(
-    "stabilityai/stable-diffusion-xl-base-1.0",
-    torch_dtype=torch.float16,
-    variant="fp16",
-    use_safetensors=True,
-)
-base.to('cuda')
-
-refiner = DiffusionPipeline.from_pretrained(
-    "stabilityai/stable-diffusion-xl-base-1.0",
-    text_encoder_2=base.text_encoder_2,
-    vae=base.vae,
-    torch_dtype=torch.float16,
-    use_safetensors=True,
-    variant="fp16",
-)
-refiner.to('cuda')
+# # Initialize Stable Diffusion pipelines
+# base = DiffusionPipeline.from_pretrained(
+#     "stabilityai/stable-diffusion-xl-base-1.0",
+#     torch_dtype=torch.float16,
+#     variant="fp16",
+#     use_safetensors=True,
+# )
+# base.to('cuda')
+
+# refiner = DiffusionPipeline.from_pretrained(
+#     "stabilityai/stable-diffusion-xl-base-1.0",
+#     text_encoder_2=base.text_encoder_2,
+#     vae=base.vae,
+#     torch_dtype=torch.float16,
+#     use_safetensors=True,
+#     variant="fp16",
+# )
+# refiner.to('cuda')
+
+# load both base and refiner
+base = DiffusionPipeline.from_pretrained("stabilityai/stable-diffusion-xl-base-1.0", torch_dtype=torch.float16, use_safetensors=True, variant="fp16").to('cuda')
+refiner = DiffusionPipeline.from_pretrained("stabilityai/stable-diffusion-xl-refiner-1.0",
+                                            text_encoder_2=base.text_encoder_2,
+                                            vae=base.vae,
+                                            torch_dtype=torch.float16,
+                                            use_safetensor=True,
+                                            variant="fp16").to('cuda')
 
 class ModeManager:
     def __init__(self):