Unverified Commit b2fb53d4 authored by Hailey Schoelkopf's avatar Hailey Schoelkopf Committed by GitHub
Browse files

Update model.py

parent f95b160b
...@@ -180,7 +180,7 @@ class CachingLM: ...@@ -180,7 +180,7 @@ class CachingLM:
remaining_reqs = [] remaining_reqs = []
warned = False warned = False
# figure out which ones are cached and which ones are new # figure out which ones are cached and which ones are new
print(f"Loading responses from cache '{self.cache_db}' where possible") print(f"Loading '{attr}' responses from cache '{self.cache_db}' where possible")
for req in tqdm(requests): for req in tqdm(requests):
hsh = hash_args(attr, req.args) hsh = hash_args(attr, req.args)
if attr == "greedy_until" and req.args[1].get("do_sample", False): if attr == "greedy_until" and req.args[1].get("do_sample", False):
......
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