Update app.py

app.py

@@ -19,7 +19,7 @@ vae = AutoencoderKL.from_pretrained("madebyollin/sdxl-vae-fp16-fix", torch_dtype
 pipe = DiffusionPipeline.from_pretrained(
     "stabilityai/stable-diffusion-xl-base-1.0",
     vae=vae, torch_dtype=torch.float16, variant="fp16",
-    #use_safetensors=True
+    use_safetensors=True
 )
 
 device="cuda" if torch.cuda.is_available() else "cpu"
@@ -70,7 +70,7 @@ def load_model(custom_model):
     return gr.update(choices=sfts_available_files, value=sfts_available_files[0], visible=True), gr.update(value=instance_prompt, visible=True)
 
 
-@spaces.GPU
+
 def infer (custom_model, weight_name, prompt, inf_steps, guidance_scale, seed, lora_weight, progress=gr.Progress(track_tqdm=True)):
 
     if weight_name == "NO SAFETENSORS FILE":