using model "GuangyuanSD/Z-Image-Re-Turbo-LoRA"

app.py

@@ -6,22 +6,13 @@ import random
 from diffusers import DiffusionPipeline
 import torch
 
-device = "cuda" if torch.cuda.is_available() else "cpu"
-model_repo_id = "stabilityai/sdxl-turbo"  # Replace to the model you would like to use
-
-if torch.cuda.is_available():
-    torch_dtype = torch.float16
-else:
-    torch_dtype = torch.float32
-
-pipe = DiffusionPipeline.from_pretrained(model_repo_id, torch_dtype=torch_dtype)
-pipe = pipe.to(device)
+pipe = DiffusionPipeline.from_pretrained("GuangyuanSD/Z-Image-Re-Turbo-LoRA", dtype=torch.bfloat16, device_map="cuda")
 
 MAX_SEED = np.iinfo(np.int32).max
 MAX_IMAGE_SIZE = 1024
 
 
-# @spaces.GPU #[uncomment to use ZeroGPU]
+@spaces.GPU
 def infer(
     prompt,
     negative_prompt,