Update app.py

app.py

@@ -7,23 +7,21 @@ from diffusers import DiffusionPipeline, FlowMatchEulerDiscreteScheduler
 import torch
 
 device = "cuda" if torch.cuda.is_available() else "cpu"
-model_repo_id = "tensorart/stable-diffusion-3.5-large-TurboX"
+device = "cuda" if torch.cuda.is_available() else "cpu"
+model_repo_id = "stabilityai/stable-diffusion-3.5-large-turbo"
 
 if torch.cuda.is_available():
-    torch_dtype = torch.float16
+    torch_dtype = torch.bfloat16
 else:
     torch_dtype = torch.float32
 
 pipe = DiffusionPipeline.from_pretrained(model_repo_id, torch_dtype=torch_dtype)
-
-pipe.scheduler = FlowMatchEulerDiscreteScheduler.from_pretrained(model_repo_id, subfolder="scheduler", shift=5)
-
 pipe = pipe.to(device)
 
 MAX_SEED = np.iinfo(np.int32).max
 MAX_IMAGE_SIZE = 2048
 
-@spaces.GPU(duration=25)
+@spaces.GPU(duration=20)
 def infer(
     prompt,
     negative_prompt="",