Update skip compile message

Signed-off-by: Andrea Fasoli <andrea.fasoli@ibm.com>

Author: andrea-fasoli

scripts/run_decoder.py

@@ -8,7 +8,6 @@
 from torch import distributed, set_grad_enabled
 
 # Local Packages
-from aiu_fms_testing_utils.utils import aiu_setup, warmup_model
 from aiu_fms_testing_utils.utils.aiu_setup import dprint, rank
 from aiu_fms_testing_utils.utils.args_parsing import get_args
 from aiu_fms_testing_utils.utils.decoders_utils import run_decoder_eval
@@ -80,7 +79,7 @@
         model.compile(mode=args.compile_mode, backend=args.compile_backend)
     dprint("Model compiled.")
 else:
-    dprint("[WARNING] SKIP COMPILE.")
+    dprint("Skip model compiling. Only for debug purpose.")
 
 run_decoder_eval(model, tokenizer, args, device)
 

scripts/run_encoder.py

@@ -42,6 +42,7 @@
     save_path = None
 
     # !!! insert DQ for encoders here
+    # pass default_dtype to DQ function
 
     # if DQ is used, args.model_path represent FP16 ckpt but we need to load the
     # newly-created INT8 ckpt. Without DQ, args.model_path is the INT8 ckpt already.
@@ -96,7 +97,7 @@
         model.compile(mode=args.compile_mode, backend=args.compile_backend)
     dprint("Model compiled.")
 else:
-    dprint("[WARNING] SKIP COMPILE.")
+    dprint("Skip model compiling. Only for debug purpose.")
 
 if args.architecture == "roberta_question_answering":
     run_encoder_eval_qa(model, tokenizer, args)