Spaces:

alexkueck
/

ChatBotLI2Klein

Paused

alexkueck commited on Jul 1, 2023

Commit

d941c89

1 Parent(s): 2f393e0

Update utils.py

Files changed (1) hide show

utils.py CHANGED Viewed

@@ -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
 def reset_state():

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