Update app.py

app.py

@@ -21,7 +21,7 @@ args = OmegaConf.load("inference_configs/inference.yaml")
 # is_schnell = args.model_name == "flux-schnell"
 # sampler = None
 device = torch.device("cuda")
-# dtype = torch.bfloat16
+dtype = torch.bfloat16
 # dit = load_flow_model2(args.model_name, device="cpu").to(device, dtype=dtype)
 # vae = load_ae(args.model_name, device="cpu").to(device, dtype=dtype)
 # t5 = load_t5(device="cpu", max_length=256 if is_schnell else 512).to(device, dtype=dtype)