Update app.py

app.py

@@ -71,11 +71,11 @@ refiner = StableDiffusionXLImg2ImgPipeline.from_pretrained("ford442/stable-diffu
 #refiner = StableDiffusionXLImg2ImgPipeline.from_pretrained("stabilityai/stable-diffusion-xl-refiner-1.0", vae=vae, torch_dtype=torch.float32, requires_aesthetics_score=True, device_map='balanced')
 
 #refiner.enable_model_cpu_offload()
+refiner.scheduler = EulerAncestralDiscreteScheduler.from_config(refiner.scheduler.config, beta_schedule="scaled_linear")
 
 #refiner.scheduler.config.requires_aesthetics_score=False
 refiner.to(device)
 #refiner = torch.compile(refiner)
-refiner.scheduler = EulerAncestralDiscreteScheduler.from_config(refiner.scheduler.config, beta_schedule="scaled_linear")
 
 tokenizer = AutoTokenizer.from_pretrained(checkpoint, add_prefix_space=False, device_map='balanced')
 tokenizer.tokenizer_legacy=False
@@ -168,6 +168,7 @@ def infer(
     image_path = f"sd35m_{seed}.png"
     sd_image.save(image_path,optimize=False,compress_level=0)
     upload_to_ftp(image_path)
+    refiner.scheduler.set_timesteps(num_inference_steps,device)
     refine = refiner(
             prompt=f"{prompt}, high quality masterpiece, complex details",
             negative_prompt = negative_prompt,