Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
|
@@ -14,8 +14,6 @@ pipeline = StableDiffusionPanoramaPipeline.from_pretrained(
|
|
| 14 |
).to("cuda")
|
| 15 |
pipeline.enable_model_cpu_offload()
|
| 16 |
pipeline.enable_vae_slicing()
|
| 17 |
-
|
| 18 |
-
|
| 19 |
sa_args = sa_handler.StyleAlignedArgs(share_group_norm=True,
|
| 20 |
share_layer_norm=True,
|
| 21 |
share_attention=True,
|
|
@@ -27,7 +25,6 @@ handler = sa_handler.Handler(pipeline)
|
|
| 27 |
handler.register(sa_args)
|
| 28 |
|
| 29 |
|
| 30 |
-
|
| 31 |
# run MultiDiffusion with StyleAligned
|
| 32 |
def style_aligned_multidiff(ref_style_prompt, img_generation_prompt):
|
| 33 |
view_batch_size = 25 # adjust according to VRAM size
|
|
|
|
| 14 |
).to("cuda")
|
| 15 |
pipeline.enable_model_cpu_offload()
|
| 16 |
pipeline.enable_vae_slicing()
|
|
|
|
|
|
|
| 17 |
sa_args = sa_handler.StyleAlignedArgs(share_group_norm=True,
|
| 18 |
share_layer_norm=True,
|
| 19 |
share_attention=True,
|
|
|
|
| 25 |
handler.register(sa_args)
|
| 26 |
|
| 27 |
|
|
|
|
| 28 |
# run MultiDiffusion with StyleAligned
|
| 29 |
def style_aligned_multidiff(ref_style_prompt, img_generation_prompt):
|
| 30 |
view_batch_size = 25 # adjust according to VRAM size
|