Spaces:
Runtime error
Runtime error
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 |
|