TypeError: LLaDAModelLM.__init__() got an unexpected keyword argument 'use_flash_attn'

handler.py

@@ -9,7 +9,6 @@ class EndpointHandler:
             model_dir,
             torch_dtype=torch.bfloat16,
             low_cpu_mem_usage=True,
-            use_flash_attn=False,
             trust_remote_code=True,
             device_map="auto",
         ).eval()