Procházet zdrojové kódy

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)

DooWoong Lee (David) před 2 roky
rodič
revize
692ce3164e
1 změnil soubory, kde provedl 0 přidání a 2 odebrání
  1. 0 2
      convert-pth-to-ggml.py

+ 0 - 2
convert-pth-to-ggml.py

@@ -145,13 +145,11 @@ def main():
 
 
         print(f"Extracting only the vocab from '{fname_model}'\n")
         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:
         with open(fname_out, "wb") as fout:
             write_header(fout, hparams, ftype)
             write_header(fout, hparams, ftype)
             write_tokens(fout, tokenizer)
             write_tokens(fout, tokenizer)
 
 
-        del model
 
 
         print(f"Done. Output file: {fname_out}\n")
         print(f"Done. Output file: {fname_out}\n")