Update app.py

app.py

@@ -26,9 +26,6 @@ MAX_IMAGE_SIZE = int(os.getenv("MAX_IMAGE_SIZE", "4192"))
 USE_TORCH_COMPILE = os.getenv("USE_TORCH_COMPILE", "0") == "1"
 ENABLE_CPU_OFFLOAD = os.getenv("ENABLE_CPU_OFFLOAD", "0") == "1"
 
-device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
-
-
 style_list = [
     {
         "name": "(No style)",
@@ -81,7 +78,8 @@ NUM_IMAGES_PER_PROMPT = 1
 pipe = PixArtAlphaPipeline.from_pretrained(
         "PixArt-alpha/PixArt-LCM-XL-2-1024-MS",
         torch_dtype=torch.float16,
-        use_safetensors=True).to(device)
+        use_safetensors=True,
+).to("cuda:0")
 
 pipe.vae = ConsistencyDecoderVAE.from_pretrained("openai/consistency-decoder", torch_dtype=torch.float16)
 pipe.text_encoder.to_bettertransformer()