Update app.py

app.py

@@ -73,7 +73,7 @@ except FileNotFoundError as e:
     print(f"Error loading checkpoint: {e}\nPlease ensure the checkpoint '{args.checkpoint_path}' exists.")
     exit(1)
 
-text_encoder.eval().to(dtype=torch.bfloat16).requires_grad_(False)
+text_encoder.eval().to(dtype=torch.float16).requires_grad_(False)
 transformer.eval().to(dtype=torch.float16).requires_grad_(False)
 
 text_encoder.to(gpu)