use gpt2-medium instead of gpt2-large
Browse files
app.py
CHANGED
|
@@ -5,7 +5,7 @@ import numpy as np
|
|
| 5 |
from transformers_cfg.grammar_utils import IncrementalGrammarConstraint
|
| 6 |
from transformers_cfg.generation.logits_process import GrammarConstrainedLogitsProcessor
|
| 7 |
|
| 8 |
-
MODEL_NAME = "gpt2-
|
| 9 |
|
| 10 |
if __name__ == "__main__":
|
| 11 |
# Define your model and your tokenizer
|
|
|
|
| 5 |
from transformers_cfg.grammar_utils import IncrementalGrammarConstraint
|
| 6 |
from transformers_cfg.generation.logits_process import GrammarConstrainedLogitsProcessor
|
| 7 |
|
| 8 |
+
MODEL_NAME = "gpt2-medium"
|
| 9 |
|
| 10 |
if __name__ == "__main__":
|
| 11 |
# Define your model and your tokenizer
|