From e5e2fb088ad95ffaf69f0d4909971861765e572a Mon Sep 17 00:00:00 2001
From: Gnome Ann <>
Date: Fri, 26 Nov 2021 12:38:52 -0500
Subject: [PATCH] Remember to actually import `GPTJModel`

---
 aiserver.py | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/aiserver.py b/aiserver.py
index a379150e..51faa597 100644
--- a/aiserver.py
+++ b/aiserver.py
@@ -542,6 +542,10 @@ if(not vars.model in ["InferKit", "Colab", "OAI", "ReadOnly", "TPUMeshTransforme
     if(not vars.noai):
         print("{0}Initializing transformers, please wait...{1}".format(colors.PURPLE, colors.END))
         from transformers import StoppingCriteria, GPT2Tokenizer, GPT2LMHeadModel, GPTNeoForCausalLM, GPTNeoModel, AutoModelForCausalLM
+        try:
+            from transformers import GPTJModel
+        except:
+            pass
         import transformers.generation_utils
         from transformers import __version__ as transformers_version