Spaces:
Runtime error
Runtime error
Commit
•
0e4456d
1
Parent(s):
3eb0541
change to base
Browse files
config.py
CHANGED
|
@@ -1,6 +1,8 @@
|
|
| 1 |
class Config:
|
| 2 |
-
model = 'flan-small'
|
| 3 |
-
tokenizer = 'CreatorFPT/T5-base'
|
|
|
|
|
|
|
| 4 |
seq_len = 512
|
| 5 |
out_len = 170
|
| 6 |
device = 'cpu'
|
|
|
|
| 1 |
class Config:
|
| 2 |
+
# model = 'flan-small'
|
| 3 |
+
# tokenizer = 'CreatorFPT/T5-base'
|
| 4 |
+
model = 'vietAI-base-MRC'
|
| 5 |
+
tokenizer = 'VietAI/vit5-base'
|
| 6 |
seq_len = 512
|
| 7 |
out_len = 170
|
| 8 |
device = 'cpu'
|