Update app.py

app.py

@@ -5,14 +5,15 @@ import random
 import functools
 import torch
 from diffusers import StableDiffusion3Pipeline
+from diffusers import DiffusionPipeline
 from inference import run
 from peft import LoraConfig, get_peft_model, PeftModel
 
-pipe = StableDiffusion3Pipeline.from_pretrained("stabilityai/stable-diffusion-3.5-medium",
-                                                torch_dtype=torch.float16)
-pipe = pipe.to("cuda")
+model_repo_id = "stabilityai/stable-diffusion-3.5-large"
+pipe = DiffusionPipeline.from_pretrained(model_repo_id, torch_dtype=torch.bfloat16)
+pipe = pipe.to('cuda')
 
-distill_check = 'yresearch/swd-medium-6-steps'
+distill_check = 'yresearch/swd-large-6-steps'
 pipe.transformer = PeftModel.from_pretrained(
     pipe.transformer,
     distill_check,