Spaces:

fffiloni
/

StyleAligned_Transfer

Running on Zero

fffiloni commited on Dec 12, 2023

Commit

0473e4c

1 Parent(s): 8cb5b15

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -13,6 +13,9 @@ pipeline = StableDiffusionXLPipeline.from_pretrained(
     scheduler=scheduler
 ).to("cuda")
 handler = sa_handler.Handler(pipeline)
 sa_args = sa_handler.StyleAlignedArgs(share_group_norm=False,
                                       share_layer_norm=False,

     scheduler=scheduler
 ).to("cuda")
+pipeline.enable_model_cpu_offload()
+pipeline.enable_vae_slicing()
 handler = sa_handler.Handler(pipeline)
 sa_args = sa_handler.StyleAlignedArgs(share_group_norm=False,
                                       share_layer_norm=False,