Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -31,7 +31,6 @@ pipe_lightning = StableDiffusionXLPipeline.from_pretrained(
|
|
31 |
tokenizer=pipe_turbo.tokenizer,
|
32 |
tokenizer_2=pipe_turbo.tokenizer_2,
|
33 |
torch_dtype=torch.float16,
|
34 |
-
variant="fp16"
|
35 |
)#.to("cuda")
|
36 |
del unet
|
37 |
pipe_lightning.scheduler = EulerDiscreteScheduler.from_config(pipe_lightning.scheduler.config, timestep_spacing="trailing", prediction_type="sample")
|
|
|
31 |
tokenizer=pipe_turbo.tokenizer,
|
32 |
tokenizer_2=pipe_turbo.tokenizer_2,
|
33 |
torch_dtype=torch.float16,
|
|
|
34 |
)#.to("cuda")
|
35 |
del unet
|
36 |
pipe_lightning.scheduler = EulerDiscreteScheduler.from_config(pipe_lightning.scheduler.config, timestep_spacing="trailing", prediction_type="sample")
|