Update app.py

app.py

@@ -7,6 +7,7 @@ from diffusers import DiffusionPipeline, StableDiffusion3Pipeline
 from huggingface_hub import hf_hub_download
 
 device = 'cuda' if torch.cuda.is_available() else 'cpu'
+PYTORCH_CUDA_ALLOC_CONF=expandable_segments:True
 torch.cuda.max_memory_allocated(device=device)
 torch.cuda.empty_cache()
 pipe = DiffusionPipeline.from_pretrained("circulus/canvers-fusionXL-v1").to(device)