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

remove stray print in AnthropicLM, add TODO

parent 360f7c3e
......@@ -26,7 +26,6 @@ def anthropic_completion(
max_tokens_to_sample=max_tokens_to_sample,
temperature=temperature,
)
print(response)
return response["completion"]
except RuntimeError:
# TODO: I don't actually know what error Anthropic raises when it times out
......@@ -99,7 +98,7 @@ class AnthropicLM(LM):
model=self.model,
prompt=inp,
max_tokens_to_sample=self.max_gen_toks,
temperature=0.0,
temperature=0.0, # TODO: implement non-greedy sampling for Anthropic
stop=until,
)
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