Update app.py

app.py

@@ -267,7 +267,7 @@ device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
 
 #accelerator = Accelerator(mixed_precision="bf16") # Example
 
-upscaler = UpscaleWithModel.from_pretrained("Kim2091/ClearRealityV1").to(torch.device("cuda:0"))
+upscaler = UpscaleWithModel.from_pretrained("Kim2091/ClearRealityV1").to(device)
 
 def load_and_prepare_model():
     #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)
@@ -290,7 +290,8 @@ def load_and_prepare_model():
     print(f'init noise scale: {pipe.scheduler.init_noise_sigma}')
     pipe.watermark=None
     pipe.safety_checker=None  
-    pipe.to(torch.device('cuda:0'), torch.bfloat16)
+    #pipe.to(torch.device('cuda:0'), torch.bfloat16)
+    pipe.to(torch.float16)
     pipe.vae = vaeXL #.to(torch.bfloat16)
     #pipe.to(torch.bfloat16)
     #pipe.to(accelerator.device)
@@ -318,7 +319,7 @@ def load_and_prepare_model():
 pipe = load_and_prepare_model()
 neg_prompt_2 = " 'non-photorealistic':1.5, 'unrealistic skin','unattractive face':1.3, 'low quality':1.1, ('dull color scheme', 'dull colors', 'digital noise':1.2),'amateurish', 'poorly drawn face':1.3, 'poorly drawn', 'distorted face', 'low resolution', 'simplistic' "
 
-@spaces.GPU(duration=40)
+#@spaces.GPU(duration=40)
 def generate_30(
     prompt: str,
     negative_prompt: str = "",
@@ -376,7 +377,7 @@ def generate_30(
     os.symlink(sd_image_path, unique_name)  
     return [unique_name]
 
-@spaces.GPU(duration=70)
+#@spaces.GPU(duration=70)
 def generate_60(
     prompt: str,
     negative_prompt: str = "",
@@ -434,7 +435,7 @@ def generate_60(
     os.symlink(sd_image_path, unique_name)  
     return [unique_name]
     
-@spaces.GPU(duration=100)
+#@spaces.GPU(duration=100)
 def generate_90(
     prompt: str,
     negative_prompt: str = "",