Remember to actually import GPTJModel

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