Update app.py

app.py

@@ -101,7 +101,6 @@ def apply_style(style_name: str, positive: str, negative: str = "") -> Tuple[str
     return p.replace("{prompt}", positive), n + negative
 
 def load_and_prepare_model():
-    torch.set_default_device('cuda')
     #vae = AutoencoderKL.from_pretrained("ford442/sdxl-vae-bf16", safety_checker=None)
     vaeX = AutoencoderKL.from_pretrained("stabilityai/sdxl-vae", safety_checker=None,use_safetensors=False)
     #vae = AutoencoderKL.from_single_file('https://huggingface.co/ford442/sdxl-vae-bf16/mySLR/myslrVAE_v10.safetensors')
@@ -260,7 +259,7 @@ def generate_30(
     lora_scale: float = 0.5,
     progress=gr.Progress(track_tqdm=True)  # Add progress as a keyword argument
 ):
-    #torch.backends.cudnn.benchmark = False
+    torch.set_default_device('cuda')
     pipe.set_adapters(["skin"], adapter_weights=[lora_scale])
     seed = int(randomize_seed_fn())
     generator = torch.Generator(device='cuda').manual_seed(seed)
@@ -318,7 +317,7 @@ def generate_60(
     lora_scale: float = 0.5,
     progress=gr.Progress(track_tqdm=True)  # Add progress as a keyword argument
 ):
-    #torch.backends.cudnn.benchmark = False
+    torch.set_default_device('cuda')
     pipe.set_adapters(["skin"], adapter_weights=[lora_scale])
     seed = int(randomize_seed_fn())
     generator = torch.Generator(device='cuda').manual_seed(seed)
@@ -376,7 +375,7 @@ def generate_90(
     lora_scale: float = 0.5,
     progress=gr.Progress(track_tqdm=True)  # Add progress as a keyword argument
 ):
-    #torch.backends.cudnn.benchmark = False
+    torch.set_default_device('cuda')
     pipe.set_adapters(["skin"], adapter_weights=[lora_scale])
     seed = int(randomize_seed_fn())
     generator = torch.Generator(device='cuda').manual_seed(seed)