fix: dtype

app.py

@@ -14,8 +14,8 @@ from utils.sd3 import sd3_editing
 
 device = "cuda" if torch.cuda.is_available() else "cpu"
 
-pipe_sd3 = StableDiffusion3Pipeline.from_pretrained("stabilityai/stable-diffusion-3-medium-diffusers", dtype=torch.float16, token=os.getenv('HF_ACCESS_TOK'))
-pipe_flux = FluxPipeline.from_pretrained("black-forest-labs/FLUX.1-dev", dtype=torch.float16, token=os.getenv('HF_ACCESS_TOK'))
+pipe_sd3 = StableDiffusion3Pipeline.from_pretrained("stabilityai/stable-diffusion-3-medium-diffusers", torch_dtype=torch.float16, token=os.getenv('HF_ACCESS_TOK'))
+pipe_flux = FluxPipeline.from_pretrained("black-forest-labs/FLUX.1-dev", torch_dtype=torch.float16, token=os.getenv('HF_ACCESS_TOK'))
 
 
 def seed_everything(seed: int) -> None: