Spaces:

hideosnes
/

SDXL-Lightning

Paused

hideosnes commited on May 5, 2024

Commit

f6debc2

verified ·

1 Parent(s): 1b581af

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -43,7 +43,7 @@ pipe = StableDiffusionXLControlNetPipeline.from_pretrained(
     torch_dtype=torch.float16,
     variant="fp16",
     add_watermark=False,
-)to(device)
 pipe.set_progress_bar_config(disable=True)
 pipe.scheduler = EulerDiscreteScheduler.from_config(
     pipe.scheduler.config, timestep_spacing="trailing", prediction_type="epsilon"

     torch_dtype=torch.float16,
     variant="fp16",
     add_watermark=False,
+).to(device)
 pipe.set_progress_bar_config(disable=True)
 pipe.scheduler = EulerDiscreteScheduler.from_config(
     pipe.scheduler.config, timestep_spacing="trailing", prediction_type="epsilon"