Skip to content

Commit 355fbab

Browse files
authored
gpt-2 : remove GPT-J unnecessary import (leejet#91)
1 parent f0565d9 commit 355fbab

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

examples/gpt-2/convert-cerebras-to-ggml.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -10,7 +10,7 @@
1010
import numpy as np
1111
import re
1212

13-
from transformers import GPTJForCausalLM, AutoModelForCausalLM
13+
from transformers import AutoModelForCausalLM
1414

1515
# ref: https://github.com/openai/gpt-2/blob/master/src/encoder.py
1616
def bytes_to_unicode():

0 commit comments

Comments
 (0)