Remove debug lines

octoml · Feb 21, 2024 · a4bda30 · a4bda30
1 parent bf0c848
commit a4bda30
Show file tree

Hide file tree

Showing 2 changed files with 0 additions and 2 deletions.
diff --git a/serve/tests/unittest/test_engine_with_samplers.py b/serve/tests/unittest/test_engine_with_samplers.py
@@ -218,7 +218,6 @@ def _test_logit_bias(
 ):
     prompt = "Repeat only one of the following words: hi, hello"
     requests = []
-    print(engine.tokenizer.encode("hi"))
     for n in range(num_requests):
         requests.append(
             create_request(

diff --git a/serve/tests/unittest/test_sampler.py b/serve/tests/unittest/test_sampler.py
@@ -259,7 +259,6 @@ def _get_expected_result(
                 - mask[i] * presence_penalties[i])
                 / temperature
             )
-            print("DEBUG")
         return expected
 
     vocab_size = 512