Spaces:
Paused
Paused
Update utils.py
Browse files
utils.py
CHANGED
@@ -18,7 +18,7 @@ from pygments import highlight
|
|
18 |
from pygments.lexers import guess_lexer,get_lexer_by_name
|
19 |
from pygments.formatters import HtmlFormatter
|
20 |
import transformers
|
21 |
-
from transformers import AutoTokenizer, AutoModelForCausalLM
|
22 |
import datasets
|
23 |
from datasets import load_dataset
|
24 |
import evaluate
|
|
|
18 |
from pygments.lexers import guess_lexer,get_lexer_by_name
|
19 |
from pygments.formatters import HtmlFormatter
|
20 |
import transformers
|
21 |
+
from transformers import AutoTokenizer, AutoModelForCausalLM, BitsAndBytesConfig,
|
22 |
import datasets
|
23 |
from datasets import load_dataset
|
24 |
import evaluate
|