Spaces:
Running on Zero
Running on Zero
Update generator.py
Browse files- generator.py +1 -1
generator.py
CHANGED
|
@@ -6,7 +6,7 @@ Switch GENERATOR_MODEL in config.py to use Mistral, Gemma, BioMistral, etc.
|
|
| 6 |
"""
|
| 7 |
|
| 8 |
from transformers import pipeline, AutoTokenizer, AutoModelForSeq2SeqLM
|
| 9 |
-
from
|
| 10 |
|
| 11 |
_generator = None # lazy singleton
|
| 12 |
|
|
|
|
| 6 |
"""
|
| 7 |
|
| 8 |
from transformers import pipeline, AutoTokenizer, AutoModelForSeq2SeqLM
|
| 9 |
+
from config import GENERATOR_MODEL, MAX_NEW_TOKENS, TEMPERATURE
|
| 10 |
|
| 11 |
_generator = None # lazy singleton
|
| 12 |
|