[Precision] Change lm_head layer running in float32 (#3596)

* support lm_head fp32 bf16 fp16

* delete print

* code check

* check

* check

* code check

* check

* check

fastdeploy/worker/worker_process.py

@@ -587,6 +587,11 @@ def parse_args():
         action="store_true",
         help="Enable output of token-level log probabilities.",
     )
+    parser.add_argument(
+        "--lm_head_fp32",
+        action="store_true",
+        help="The data type of lm_head",
+    )
 
     args = parser.parse_args()
     return args