Unverified Commit dc1b8bcf authored by Ying Sheng's avatar Ying Sheng Committed by GitHub
Browse files

Format (#593)

parent 5a57b8ad
......@@ -37,10 +37,12 @@ def get_logits_batch(url, prompts):
},
)
ret = response.json()
logits = np.array(list(
ret[i]["meta_info"]["normalized_prompt_logprob"]
for i in range(len(prompts))
))
logits = np.array(
list(
ret[i]["meta_info"]["normalized_prompt_logprob"]
for i in range(len(prompts))
)
)
return logits
......@@ -64,4 +66,4 @@ if __name__ == "__main__":
"This is a long long long long test prompt.<|eot_id|>",
]
logits = get_logits_batch(url, prompts)
print(f"{logits=}")
\ No newline at end of file
print(f"{logits=}")
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