py : removed unused `model` variable and verified that the code functions correctly with `vocab_only` setting. Also confirmed that the code works as expected after running with reduced memory usage due to deletion of no-longer-needed variable. (#547)

pull/509/head^2
DooWoong Lee (David) 1 year ago committed by GitHub
parent 96f9c0506f
commit 692ce3164e
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

@ -145,13 +145,11 @@ def main():
print(f"Extracting only the vocab from '{fname_model}'\n")
model = torch.load(fname_model, map_location="cpu")
with open(fname_out, "wb") as fout:
write_header(fout, hparams, ftype)
write_tokens(fout, tokenizer)
del model
print(f"Done. Output file: {fname_out}\n")

Loading…
Cancel
Save