Update app.py

app.py

@@ -8,8 +8,6 @@ import time
 import psutil
 import random
 
-CUDA_LAUNCH_BLOCKING=1
-
 start_time = time.time()
 is_colab = utils.is_google_colab()
 state = None
@@ -88,19 +86,10 @@ else:
       torch_dtype=torch.get_default_dtype(),
       scheduler=DPMSolverMultistepScheduler.from_pretrained(current_model.path, subfolder="scheduler")
       )
-    
-if torch.cuda.is_available():
-  pipe = pipe.to("cuda")
-  pipe.enable_xformers_memory_efficient_attention()
 
-def to_cuda(torch, pipe):
-    try:
         if torch.cuda.is_available():
           pipe = pipe.to("cuda")
           pipe.enable_xformers_memory_efficient_attention()
-        return True
-    except:
-        return False
 
 device = "GPU 🔥" if torch.cuda.is_available() else "CPU 🥶"
 
@@ -188,16 +177,9 @@ def txt_to_img(model_path, prompt, n_images, neg_prompt, guidance, steps, width,
           # pipe = pipe.to("cpu")
           # pipe = current_model.pipe_t2i
 
-to_cuda(torch, pipe)
-
-def to_cuda(torch, pipe):
-    try:
         if torch.cuda.is_available():
           pipe = pipe.to("cuda")
           pipe.enable_xformers_memory_efficient_attention()
-        return True
-    except:
-        return False
         last_mode = "txt2img"
 
     prompt = current_model.prefix + prompt  
@@ -244,16 +226,9 @@ def img_to_img(model_path, prompt, n_images, neg_prompt, img, strength, guidance
           # pipe = pipe.to("cpu")
           # pipe = current_model.pipe_i2i
 
-to_cuda(torch, pipe)
-
-def to_cuda(torch, pipe):
-    try:
         if torch.cuda.is_available():
           pipe = pipe.to("cuda")
           pipe.enable_xformers_memory_efficient_attention()
-        return True
-    except:
-        return False
         last_mode = "img2img"
 
     prompt = current_model.prefix + prompt