diff --git a/aiserver.py b/aiserver.py index c5f8c1ec..35131612 100644 --- a/aiserver.py +++ b/aiserver.py @@ -1063,10 +1063,10 @@ if(not vars.model in ["InferKit", "Colab", "OAI", "ReadOnly", "TPUMeshTransforme if not args.colab: import shutil - shutil.rmtree("cache/") model = model.half() model.save_pretrained("models/{}".format(vars.model.replace('/', '_'))) tokenizer.save_pretrained("models/{}".format(vars.model.replace('/', '_'))) + shutil.rmtree("cache/") if(vars.hascuda): if(vars.usegpu):