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")
|