Update app.py

app.py

@@ -5,6 +5,7 @@ import random
 import spaces
 from diffusers import DiffusionPipeline, FlowMatchEulerDiscreteScheduler
 import torch
+torch.cuda.empty_cache()
 
 device = "cuda" if torch.cuda.is_available() else "cpu"
 model_repo_id = "tensorart/stable-diffusion-3.5-large-TurboX"
@@ -14,9 +15,9 @@ if torch.cuda.is_available():
 else:
     torch_dtype = torch.bfloat16
 
-pipe = DiffusionPipeline.from_pretrained(model_repo_id, torch_dtype=torch_dtype, use_fast=True)
+    pipe = DiffusionPipeline.from_pretrained(model_repo_id, torch_dtype=torch_dtype)
 
-pipe.scheduler = FlowMatchEulerDiscreteScheduler.from_pretrained(model_repo_id, subfolder="scheduler", shift=5)
+pipe.scheduler = FlowMatchEulerDiscreteScheduler.from_pretrained(model_repo_id, subfolder="scheduler", shift=5, use_fast=True)
 
 pipe = pipe.to(device)