alexkueck commited on
Commit
8d403b4
1 Parent(s): 4c85a93

Update utils.py

Browse files
Files changed (1) hide show
  1. utils.py +2 -2
utils.py CHANGED
@@ -20,8 +20,8 @@ from pygments.lexers import guess_lexer,get_lexer_by_name
20
  from pygments.formatters import HtmlFormatter
21
  import transformers
22
  from transformers import AutoTokenizer, AutoModelForCausalLM, GPT2Tokenizer, GPT2LMHeadModel
23
- import auto-gptq
24
- from auto-gptq import AutoGPTQForCausalLM, BaseQuantizeConfig
25
  from transformers import LlamaForCausalLM, LlamaTokenizer
26
 
27
 
 
20
  from pygments.formatters import HtmlFormatter
21
  import transformers
22
  from transformers import AutoTokenizer, AutoModelForCausalLM, GPT2Tokenizer, GPT2LMHeadModel
23
+ #import auto-gptq
24
+ #from auto-gptq import AutoGPTQForCausalLM, BaseQuantizeConfig
25
  from transformers import LlamaForCausalLM, LlamaTokenizer
26
 
27