gpt-2 : remove GPT-J unnecessary import (#91)

diff --git a/examples/gpt-2/convert-cerebras-to-ggml.py b/examples/gpt-2/convert-cerebras-to-ggml.py
index 7fba7cde7b6959cea7bbc3d5a6e7e98eefd355ee..5e7113401b517bddafeb5f91c2e7ef5e5a970575 100644 (file)
--- a/examples/gpt-2/convert-cerebras-to-ggml.py
+++ b/examples/gpt-2/convert-cerebras-to-ggml.py
@@ -10,7 +10,7 @@ import torch
 import numpy as np
 import re
 
-from transformers import GPTJForCausalLM, AutoModelForCausalLM
+from transformers import AutoModelForCausalLM
 
 # ref: https://github.com/openai/gpt-2/blob/master/src/encoder.py
 def bytes_to_unicode():