Update app.py

app.py

@@ -33,6 +33,10 @@ import datetime
 import cyper
 from PIL import Image
 
+from accelerate import Accelerator
+
+accelerator = Accelerator(mixed_precision="bf16")
+
 hftoken = os.getenv("HF_AUTH_TOKEN") 
 
 code = r'''
@@ -90,7 +94,9 @@ text_encoder_3=T5EncoderModel.from_pretrained("ford442/stable-diffusion-3.5-larg
 ll_transformer=SD3Transformer2DModel.from_pretrained("ford442/stable-diffusion-3.5-large-bf16", subfolder='transformer',token=True).to(torch.device("cuda:0"), dtype=torch.bfloat16)
 pipe.transformer=ll_transformer.eval()
 pipe.load_lora_weights("ford442/sdxl-vae-bf16", weight_name="LoRA/UltraReal.safetensors")
-pipe.to(device=device, dtype=torch.bfloat16)
+
+pipe.to(accelerator.device)
+#pipe.to(device=device, dtype=torch.bfloat16)
 
 upscaler_2 = UpscaleWithModel.from_pretrained("Kim2091/ClearRealityV1").to(torch.device('cpu'))