Skip to content

GitLab

Commit 1c409035 authored by haileyschoelkopf's avatar haileyschoelkopf Committed by lintangsutawika
Browse files

remove stray print in AnthropicLM, add TODO

parent 360f7c3e
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 2 deletions
lm_eval/models/anthropic_llms.py
View file @ 1c409035
...@@ -26,7 +26,6 @@ def anthropic_completion( ...@@ -26,7 +26,6 @@ def anthropic_completion(
max_tokens_to_sample=max_tokens_to_sample, max_tokens_to_sample=max_tokens_to_sample,
temperature=temperature, temperature=temperature,
) )
print(response)
return response["completion"] return response["completion"]
except RuntimeError: except RuntimeError:
# TODO: I don't actually know what error Anthropic raises when it times out # TODO: I don't actually know what error Anthropic raises when it times out
...@@ -99,7 +98,7 @@ class AnthropicLM(LM): ...@@ -99,7 +98,7 @@ class AnthropicLM(LM):
model=self.model, model=self.model,
prompt=inp, prompt=inp,
max_tokens_to_sample=self.max_gen_toks, max_tokens_to_sample=self.max_gen_toks,
temperature=0.0, temperature=0.0, # TODO: implement non-greedy sampling for Anthropic
stop=until, stop=until,
) )
res.append(response) res.append(response)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment