From 871ed65570f7ef94fe4a6668e2b66dec6daddd01 Mon Sep 17 00:00:00 2001 From: Gnome Ann <> Date: Fri, 26 Nov 2021 11:42:04 -0500 Subject: [PATCH] Remove an unnecessary `**maybe_low_cpu_mem_usage()` --- aiserver.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/aiserver.py b/aiserver.py index 9bb246b1..a379150e 100644 --- a/aiserver.py +++ b/aiserver.py @@ -749,7 +749,7 @@ if(not vars.model in ["InferKit", "Colab", "OAI", "ReadOnly", "TPUMeshTransforme js = json.load(model_config) with(maybe_use_float16()): model = GPT2LMHeadModel.from_pretrained(vars.custmodpth, cache_dir="cache/", **maybe_low_cpu_mem_usage()) - tokenizer = GPT2Tokenizer.from_pretrained(vars.custmodpth, cache_dir="cache/", **maybe_low_cpu_mem_usage()) + tokenizer = GPT2Tokenizer.from_pretrained(vars.custmodpth, cache_dir="cache/") vars.modeldim = get_hidden_size_from_model(model) # Is CUDA available? If so, use GPU, otherwise fall back to CPU if(vars.hascuda and vars.usegpu):