Update app.py

app.py

@@ -1,5 +1,5 @@
 #!/usr/bin/env python
-# Patch0.1x
+#Patch0.1x
 import os
 import random
 import uuid
@@ -74,17 +74,18 @@ NUM_IMAGES_PER_PROMPT = 1
 
 if torch.cuda.is_available():
     pipe = DiffusionPipeline.from_pretrained(
-        "Corcelio/mobius",
+        "yodayo-ai/kivotos-xl-2.0",
         torch_dtype=torch.float16,
         use_safetensors=True,
-        add_watermarker=False
+        add_watermarker=False,
+        variant="fp16"
     )
     if ENABLE_CPU_OFFLOAD:
         pipe.enable_model_cpu_offload()
     else:
-        pipe.to(device)
+        pipe.to(device)    
         print("Loaded on Device!")
-
+    
     if USE_TORCH_COMPILE:
         pipe.unet = torch.compile(pipe.unet, mode="reduce-overhead", fullgraph=True)
         print("Model Compiled!")