Update app.py

app.py

@@ -12,8 +12,6 @@ import torch
 from diffusers import DiffusionPipeline
 
 DESCRIPTION = '# SD-XL'
-if not torch.cuda.is_available():
-    DESCRIPTION += '\n<p>Running on CPU 🥶 This demo does not work on CPU.</p>'
 
 MAX_SEED = np.iinfo(np.int32).max
 CACHE_EXAMPLES = torch.cuda.is_available() and os.getenv(
@@ -28,8 +26,8 @@ if torch.cuda.is_available():
     pipe = DiffusionPipeline.from_pretrained(
         model,
         torch_dtype=torch.float16,
-        use_safetensors=True,
         variant='fp16')
+    pipe.load_lora_weights(model, weight_name="pytorch_lora_weights.safetensors")
     
     refiner = DiffusionPipeline.from_pretrained(
         'stabilityai/stable-diffusion-xl-refiner-1.0',