Spaces:

alexkueck
/

ChatBotLI2Klein

Paused

alexkueck commited on Sep 5, 2023

Commit

8d403b4

1 Parent(s): 4c85a93

Update utils.py

Files changed (1) hide show

utils.py CHANGED Viewed

@@ -20,8 +20,8 @@ from pygments.lexers import guess_lexer,get_lexer_by_name
 from pygments.formatters import HtmlFormatter
 import transformers
 from transformers import AutoTokenizer, AutoModelForCausalLM, GPT2Tokenizer, GPT2LMHeadModel
-import auto-gptq
-from auto-gptq import AutoGPTQForCausalLM, BaseQuantizeConfig
 from transformers import LlamaForCausalLM, LlamaTokenizer

 from pygments.formatters import HtmlFormatter
 import transformers
 from transformers import AutoTokenizer, AutoModelForCausalLM, GPT2Tokenizer, GPT2LMHeadModel
+#import auto-gptq
+#from auto-gptq import AutoGPTQForCausalLM, BaseQuantizeConfig
 from transformers import LlamaForCausalLM, LlamaTokenizer