Update app.py

app.py

@@ -77,7 +77,7 @@ class ModelWrapper:
     def _get_time():
         return time.time()
         
-    @spaces.GPU()
+    
     def sample(self, noise, unet_added_conditions, prompt_embed, fast_vae_decode):
         #alphas_cumprod = self.scheduler.alphas_cumprod.to(self.device)
         print("sampling...")
@@ -161,7 +161,7 @@ class ModelWrapper:
 
         return output_image_list, f"Run successfully in {(end_time-start_time):.2f} seconds"
 
-
+@spaces.GPU()
 def get_x0_from_noise(sample, model_output, alphas_cumprod, timestep):
     alpha_prod_t = alphas_cumprod[timestep].reshape(-1, 1, 1, 1)
     beta_prod_t = 1 - alpha_prod_t