Update app.py

app.py

@@ -116,7 +116,7 @@ def load_and_prepare_model(model_id):
     #sched = EulerAncestralDiscreteScheduler.from_pretrained("SG161222/RealVisXL_V5.0", subfolder='scheduler',beta_schedule="scaled_linear", steps_offset=1,timestep_spacing="trailing"))
     #sched = EulerAncestralDiscreteScheduler.from_pretrained("SG161222/RealVisXL_V5.0", subfolder='scheduler', steps_offset=1,timestep_spacing="trailing")
     #sched = EulerAncestralDiscreteScheduler.from_pretrained('ford442/RealVisXL_V5.0_BF16', subfolder='scheduler',beta_schedule="scaled_linear", beta_start=0.00085, beta_end=0.012, steps_offset=1,use_karras_sigmas=True)
-    sched = EulerAncestralDiscreteScheduler.from_pretrained('ford442/RealVisXL_V5.0_BF16', subfolder='scheduler')
+    sched = EulerAncestralDiscreteScheduler.from_pretrained('SG161222/RealVisXL_V5.0', subfolder='scheduler')
     pipeX = StableDiffusionXLPipeline.from_pretrained("SG161222/RealVisXL_V5.0").to(torch.bfloat16)
     #pipeX = StableDiffusionXLPipeline.from_pretrained("ford442/Juggernaut-XI-v11-fp32",use_safetensors=True)
     
@@ -161,8 +161,8 @@ def load_and_prepare_model(model_id):
     print(f'watermark: {pipe.watermark}')
     print(f'image_processor: {pipe.image_processor}')
     print(f'feature_extractor: {pipe.feature_extractor}')
-    print(f'VAE: {pipe.vae}')
-    print(f'UNET: {pipe.unet}')
+    #print(f'VAE: {pipe.vae}')
+    #print(f'UNET: {pipe.unet}')
     pipe.watermark=None
     pipe.safety_checker=None  
     #pipe.to(torch.device("cuda:0"))