alexkueck commited on
Commit
f2213d2
1 Parent(s): 8dfe3d6

Update utils.py

Browse files
Files changed (1) hide show
  1. utils.py +1 -1
utils.py CHANGED
@@ -21,7 +21,7 @@ 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
 
26
 
27
  def reset_state():
 
21
  import transformers
22
  from transformers import AutoTokenizer, AutoModelForCausalLM, GPT2Tokenizer, GPT2LMHeadModel
23
  #import auto_gptq
24
+ #from auto_gptq import AutoGPTQForCausalLM, BaseQuantizeConfig
25
 
26
 
27
  def reset_state():