rixty/KoboldAI-Client
1
1
Fork 0
mirror of https://github.com/KoboldAI/KoboldAI-Client.git synced 2025-04-16 03:07:21 +02:00
Code Issues Projects Releases Wiki Activity

Fix sentencepiece model saving

Browse Source
This commit is contained in:
henk717 2022-02-23 22:04:41 +01:00
parent 6151d16df0
commit 7de3311000
1 changed files with 1 additions and 1 deletions

2
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):