From de6ea04e12de1104ffbf3a26a70b6c44852d05f4 Mon Sep 17 00:00:00 2001 From: somebody Date: Thu, 22 Sep 2022 19:14:35 -0500 Subject: [PATCH] Push it on raw_generate! --- aiserver.py | 11 ++++------- 1 file changed, 4 insertions(+), 7 deletions(-) diff --git a/aiserver.py b/aiserver.py index 8bcdf608..7d498431 100644 --- a/aiserver.py +++ b/aiserver.py @@ -4770,9 +4770,8 @@ def calcsubmit(txt): "Colab", "API", "CLUSTER", - "OAI", - "TPUMeshTransformerGPTJ", - "TPUMeshTransformerGPTNeoX" + # "TPUMeshTransformerGPTJ", + # "TPUMeshTransformerGPTNeoX" ): generate(subtxt, min, max, found_entries) elif koboldai_vars.model == "Colab": @@ -4781,10 +4780,8 @@ def calcsubmit(txt): sendtoapi(utils.decodenewlines(tokenizer.decode(subtxt)), min, max) elif koboldai_vars.model == "CLUSTER": sendtocluster(utils.decodenewlines(tokenizer.decode(subtxt)), min, max) - elif koboldai_vars.model == "OAI": - oairequest(utils.decodenewlines(tokenizer.decode(subtxt)), min, max) - elif koboldai_vars.use_colab_tpu or koboldai_vars.model in ("TPUMeshTransformerGPTJ", "TPUMeshTransformerGPTNeoX"): - tpumtjgenerate(subtxt, min, max, found_entries=found_entries) + # elif koboldai_vars.use_colab_tpu or koboldai_vars.model in ("TPUMeshTransformerGPTJ", "TPUMeshTransformerGPTNeoX"): + # tpumtjgenerate(subtxt, min, max, found_entries=found_entries) # For InferKit web API else: