Spaces:

alexeyGod
/

sdxl_turbo2

Sleeping

alexeyGod commited on Apr 26

Commit

00885b5

verified ·

1 Parent(s): 572262c

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -61,7 +61,7 @@ pipe = StableDiffusionPipeline.from_pretrained(
      model_repo_id, torch_dtype=torch_dtype, use_safetensors=True)
 #pipe.vae = AutoencoderTiny.from_pretrained(
 #    "sayakpaul/taesd-diffusers", torch_dtype=torch_dtype, use_safetensors=True)
-pipeline.scheduler = EulerDiscreteScheduler.from_config(pipeline.scheduler.config)
 pipe = pipe.to(device)
 pipe.enable_vae_tiling()

      model_repo_id, torch_dtype=torch_dtype, use_safetensors=True)
 #pipe.vae = AutoencoderTiny.from_pretrained(
 #    "sayakpaul/taesd-diffusers", torch_dtype=torch_dtype, use_safetensors=True)
+pipeline.scheduler = KDPM2AncestralDiscreteScheduler.from_config(pipeline.scheduler.config)
 pipe = pipe.to(device)
 pipe.enable_vae_tiling()