From e7aa92cd86ca496d63adcf68b8afef08784a119c Mon Sep 17 00:00:00 2001
From: henk717 <henk717@msn.com>
Date: Thu, 23 Dec 2021 17:12:42 +0100
Subject: [PATCH] Update aiserver.py

---
 aiserver.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/aiserver.py b/aiserver.py
index 05ed2771..09a465ae 100644
--- a/aiserver.py
+++ b/aiserver.py
@@ -888,7 +888,7 @@ if(not vars.model in ["InferKit", "Colab", "OAI", "ReadOnly", "TPUMeshTransforme
                        model     = GPTNeoForCausalLM.from_pretrained(vars.model.replace('/', '_'), cache_dir="cache/", **maybe_low_cpu_mem_usage())
             else:
                 print("Model does not exist locally, attempting to download from Huggingface...")
-                tokenizer = GPT2TokenizerFast.from_pretrained(vars.custmodpth, cache_dir="cache/")
+                tokenizer = GPT2TokenizerFast.from_pretrained(vars.model, cache_dir="cache/")
                 with(maybe_use_float16()):
                     tokenizer = GPT2TokenizerFast.from_pretrained(vars.model.replace('/', '_'), cache_dir="cache/")
                     try: