Update app.py

app.py

@@ -24,6 +24,7 @@ import paramiko
 import gc
 import time
 import datetime
+
 #os.system("chmod +x ./cusparselt.sh")
 #os.system("./cusparselt.sh")
 #os.system("chmod +x ./cudnn.sh")
@@ -96,6 +97,7 @@ DEFAULT_STYLE_NAME = "Style Zero"
 STYLE_NAMES = list(styles.keys())
 HF_TOKEN = os.getenv("HF_TOKEN")
 
+   
 def apply_style(style_name: str, positive: str, negative: str = "") -> Tuple[str, str]:
     if style_name in styles:
         p, n = styles.get(style_name, styles[DEFAULT_STYLE_NAME])
@@ -135,9 +137,9 @@ def load_and_prepare_model(model_id):
      #  vae=AutoencoderKL.from_pretrained("BeastHF/MyBack_SDXL_Juggernaut_XL_VAE/MyBack_SDXL_Juggernaut_XL_VAE_V10(version_X).safetensors",repo_type='model',safety_checker=None),
      #   vae=AutoencoderKL.from_pretrained("stabilityai/sdxl-vae",repo_type='model',safety_checker=None, torch_dtype=torch.float32),
        # vae=AutoencoderKL.from_pretrained("ford442/sdxl-vae-bf16",repo_type='model',safety_checker=None),
-        vae=vae,
-        unet=pipeX.unet,
-        scheduler = sched,
+        #vae=vae,
+        #unet=pipeX.unet,
+        #scheduler = sched,
       #  scheduler = EulerAncestralDiscreteScheduler.from_config(pipeX.scheduler.config, beta_schedule="scaled_linear", beta_start=0.00085, beta_end=0.012, steps_offset=1)
         #scheduler=EulerAncestralDiscreteScheduler.from_config(pipe.scheduler.config, beta_schedule="scaled_linear", beta_start=0.00085, beta_end=0.012, steps_offset =1)
     )
@@ -145,8 +147,9 @@ def load_and_prepare_model(model_id):
     #pipe.scheduler = EulerAncestralDiscreteScheduler.from_config(pipe.scheduler.config, beta_schedule="scaled_linear", beta_start=0.00085, beta_end=0.012, steps_offset=1)
     #pipe.to('cuda')
 
-    #pipe.scheduler = sched
-    #pipe.unet=pipeX.unet
+    pipe.scheduler = sched
+    pipe.vae=vae
+    pipe.unet=pipeX.unet
    # pipe.scheduler=EulerAncestralDiscreteScheduler.from_config(pipe.scheduler.config)
     #pipe.to(dtype=torch.bfloat16)
     #pipe.unet = pipeX.unet