Update app.py

app.py

@@ -3,6 +3,7 @@ import os
 import random
 import uuid
 import gradio as gr
+import space
 import numpy as np
 import uuid
 from diffusers import PixArtAlphaPipeline, LCMScheduler
@@ -105,12 +106,6 @@ if torch.cuda.is_available():
         print("Using DALL-E 3 Consistency Decoder")
         pipe.vae = ConsistencyDecoderVAE.from_pretrained("openai/consistency-decoder", torch_dtype=torch.float16)
 
-    if ENABLE_CPU_OFFLOAD:
-        pipe.enable_model_cpu_offload()
-    else:
-        pipe.to(device)
-        print("Loaded on Device!")
-
     # speed-up T5
     pipe.text_encoder.to_bettertransformer()