Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -10,7 +10,6 @@ device = "cuda"
|
|
10 |
|
11 |
torch.cuda.max_memory_allocated(device=device)
|
12 |
pipe = StableDiffusionXLPipeline.from_pretrained("sd-community/sdxl-flash")
|
13 |
-
pipe.enable_xformers_memory_efficient_attention()
|
14 |
pipe = pipe.to(device)
|
15 |
pipe.scheduler = DPMSolverSinglestepScheduler.from_config(pipe.scheduler.config, timestep_spacing="trailing")
|
16 |
|
|
|
10 |
|
11 |
torch.cuda.max_memory_allocated(device=device)
|
12 |
pipe = StableDiffusionXLPipeline.from_pretrained("sd-community/sdxl-flash")
|
|
|
13 |
pipe = pipe.to(device)
|
14 |
pipe.scheduler = DPMSolverSinglestepScheduler.from_config(pipe.scheduler.config, timestep_spacing="trailing")
|
15 |
|