Update app.py

app.py

@@ -34,8 +34,8 @@ dtype = torch.bfloat16
 device = "cuda" if torch.cuda.is_available() else "cpu"
 
 torch.backends.cuda.matmul.allow_tf32 = True
-taef1 = AutoencoderTiny.from_pretrained("madebyollin/taef1", torch_dtype=bfloat16).to(device)
-good_vae = AutoencoderKL.from_pretrained("ostris/Flex.1-alpha", subfolder="vae", torch_dtype=bfloat16).to(device)
+taef1 = AutoencoderTiny.from_pretrained("madebyollin/taef1", torch_dtype=torch.bfloat16).to(device)
+good_vae = AutoencoderKL.from_pretrained("ostris/Flex.1-alpha", subfolder="vae", torch_dtype=torch.bfloat16).to(device)
 
 dtype = torch.bfloat16
 base_model = "AlekseyCalvin/HSTcolor_FlexSoonr"