Fix docstrings

Signed-off-by: Thara Palanivel <[email protected]>

Author: tharapalanivel

fms_mo/run_quant.py

@@ -64,11 +64,10 @@ def quantize(
             the model
         data_args (fms_mo.training_args.DataArguments): Data arguments to be used when loading the
             tokenized dataset
+        opt_args (fms_mo.training_args.OptArguments): Generic optimization related arguments
         fms_mo_args (fms_mo.training_args.FMSMOArguments): Parameters to use for PTQ quantization
         gptq_args (fms_mo.training_args.GPTQArguments): Parameters to use for GPTQ quantization
         fp8_args (fms_mo.training_args.FP8Arguments): Parameters to use for FP8 quantization
-        quant_method (str): Quantization technique, options are gptq, fp8 and dq
-        output_dir (str) Output directory to write to
     """
 
     logger.info(f"{fms_mo_args}\n{opt_args.quant_method}\n")
@@ -108,8 +107,8 @@ def run_gptq(model_args, data_args, opt_args, gptq_args):
             the model
         data_args (fms_mo.training_args.DataArguments): Data arguments to be used when loading the
             tokenized dataset
+        opt_args (fms_mo.training_args.OptArguments): Generic optimization related arguments
         gptq_args (fms_mo.training_args.GPTQArguments): Parameters to use for GPTQ quantization
-        output_dir (str) Output directory to write to
     """
 
     # Third Party
@@ -171,8 +170,8 @@ def run_fp8(model_args, data_args, opt_args, fp8_args):
             the model
         data_args (fms_mo.training_args.DataArguments): Data arguments to be used when loading the
             tokenized dataset
+        opt_args (fms_mo.training_args.OptArguments): Generic optimization related arguments
         fp8_args (fms_mo.training_args.FP8Arguments): Parameters to use for FP8 quantization
-        output_dir (str) Output directory to write to
     """
 
     # Third Party