Spaces:
Paused
Paused
Update utils.py
Browse files
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():
|