Merge pull request #13 from brianhie/remove_print

chore: remove unnecessary print statement
togethercomputer · Feb 18, 2024 · b0bb978 · b0bb978
2 parents 92a91cd + 608a5c6
commit b0bb978
Showing 1 changed file with 0 additions and 1 deletion.
diff --git a/src/model.py b/src/model.py
@@ -375,7 +375,6 @@ def stateless_forward(self, x, padding_mask=None):
         return x, None
 
     def initialize_inference_params(self):
-        print_rank_0("Initializing inference params...")
         inference_params_dict = {
             "mha": InferenceParams(
                 max_seqlen=self.config.get("max_seqlen", 8192),