Spaces:
Runtime error
Runtime error
Daniel Fried
commited on
Commit
·
bcb63b8
1
Parent(s):
460cb94
big model
Browse files- modules/app.py +1 -1
modules/app.py
CHANGED
|
@@ -9,7 +9,7 @@ if os.path.exists('use_normal_tokenizers'):
|
|
| 9 |
CUDA = False
|
| 10 |
else:
|
| 11 |
import tokenizers_patch
|
| 12 |
-
BIG_MODEL =
|
| 13 |
CUDA = True
|
| 14 |
from transformers import AutoModelForCausalLM, AutoTokenizer
|
| 15 |
import json
|
|
|
|
| 9 |
CUDA = False
|
| 10 |
else:
|
| 11 |
import tokenizers_patch
|
| 12 |
+
BIG_MODEL = True
|
| 13 |
CUDA = True
|
| 14 |
from transformers import AutoModelForCausalLM, AutoTokenizer
|
| 15 |
import json
|