Update app.py

app.py

@@ -15,7 +15,7 @@ import numpy as np
 from PIL import Image
 import torch
 #import diffusers
-from diffusers import AutoencoderKL, StableDiffusionXLPipeline
+from diffusers import AutoencoderKL, StableDiffusionXLPipeline, UNet2DConditionModel
 from diffusers import EulerAncestralDiscreteScheduler
 from typing import Tuple
 import paramiko
@@ -100,6 +100,13 @@ def load_and_prepare_model():
         add_watermarker=False,
         #text_encoder=None,
         #text_encoder_2=None,
+        unet=UNet2DConditionModel.from_pretrained(
+            'ford442/RealVisXL_V5.0_BF16',
+            low_cpu_mem_usage=False,
+            subfolder='unet',
+            upcast_attention=True,
+           # attention_type='gated-text-image',
+            token=True)
         vae=None,
     )
     #pipe.vae = vaeXL #.to(torch.bfloat16)