style-aligned_sdxl

Sleeping

ysharma HF staff commited on Dec 6, 2023

Commit

46d9ce2

•

1 Parent(s): abaacca

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -11,7 +11,8 @@ pipeline = StableDiffusionXLPipeline.from_pretrained(
     "stabilityai/stable-diffusion-xl-base-1.0", torch_dtype=torch.float16, variant="fp16", use_safetensors=True,
     scheduler=scheduler
 ).to("cuda")
-pipeline.enable_sequential_cpu_offload()
 pipeline.enable_vae_slicing()
 handler = sa_handler.Handler(pipeline)

     "stabilityai/stable-diffusion-xl-base-1.0", torch_dtype=torch.float16, variant="fp16", use_safetensors=True,
     scheduler=scheduler
 ).to("cuda")
+#pipeline.enable_sequential_cpu_offload()
+pipeline.enable_model_cpu_offload()
 pipeline.enable_vae_slicing()
 handler = sa_handler.Handler(pipeline)