Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -61,7 +61,7 @@ pipe = StableDiffusionPipeline.from_pretrained(
|
|
61 |
model_repo_id, torch_dtype=torch_dtype, use_safetensors=True)
|
62 |
#pipe.vae = AutoencoderTiny.from_pretrained(
|
63 |
# "sayakpaul/taesd-diffusers", torch_dtype=torch_dtype, use_safetensors=True)
|
64 |
-
pipe.scheduler =
|
65 |
#print(pipe.scheduler.compatibles)
|
66 |
pipe = pipe.to(device)
|
67 |
pipe.enable_vae_tiling()
|
|
|
61 |
model_repo_id, torch_dtype=torch_dtype, use_safetensors=True)
|
62 |
#pipe.vae = AutoencoderTiny.from_pretrained(
|
63 |
# "sayakpaul/taesd-diffusers", torch_dtype=torch_dtype, use_safetensors=True)
|
64 |
+
pipe.scheduler = LMSDiscreteScheduler.from_config(pipeline.scheduler.config)
|
65 |
#print(pipe.scheduler.compatibles)
|
66 |
pipe = pipe.to(device)
|
67 |
pipe.enable_vae_tiling()
|