Update utils.py

utils.py

@@ -18,7 +18,8 @@ from pygments import highlight
 from pygments.lexers import guess_lexer,get_lexer_by_name
 from pygments.formatters import HtmlFormatter
 import transformers
-from transformers import AutoTokenizer, AutoModelForCausalLM, GPT2Tokenizer, GPT2LMHeadModel, AutoGPTQForCausalLM
+from transformers import AutoTokenizer, AutoModelForCausalLM, GPT2Tokenizer, GPT2LMHeadModel
+from auto_gptq import AutoGPTQForCausalLM, BaseQuantizeConfig
 
 
 def reset_state():