Update app.py

app.py

@@ -28,7 +28,19 @@ def randomize_seed_fn(seed: int, randomize_seed: bool) -> int:
 MAX_SEED = np.iinfo(np.int32).max
 
 if not torch.cuda.is_available():
-    DESCRIPTION += "\n<p>Running on CPU 🥶 This demo may not work on CPU.</p>"
+    DESCRIPTION += "\n<p>Running on CPU 🥶</p>"
+    pipe = StableDiffusionXLPipeline.from_pretrained(
+        "fluently/Fluently-XL-Final",
+        torch_dtype=torch.float16,
+        use_safetensors=True,
+    )
+    pipe.scheduler = EulerAncestralDiscreteScheduler.from_config(pipe.scheduler.config)
+    
+    
+    pipe.load_lora_weights("ehristoforu/dalle-3-xl-v2", weight_name="dalle-3-xl-lora-v2.safetensors", adapter_name="dalle")
+    pipe.set_adapters("dalle")
+
+    pipe.enable_model_cpu_offload()
 
 MAX_SEED = np.iinfo(np.int32).max
 
@@ -48,7 +60,7 @@ if torch.cuda.is_available():
     pipe.load_lora_weights("ehristoforu/dalle-3-xl-v2", weight_name="dalle-3-xl-lora-v2.safetensors", adapter_name="dalle")
     pipe.set_adapters("dalle")
 
-    pipe.enable_model_cpu_offload()
+    pipe.to("cuda")