Spaces:
Sleeping
Sleeping
Update tools/news_reporter.py
Browse files- tools/news_reporter.py +2 -2
tools/news_reporter.py
CHANGED
|
@@ -3,8 +3,8 @@ from agent_state import AgentState
|
|
| 3 |
from transformers import pipeline, AutoTokenizer, AutoModelForCausalLM
|
| 4 |
import os
|
| 5 |
|
| 6 |
-
# MODEL_ID = "google/gemma-3n-E4B-it" #
|
| 7 |
-
MODEL_ID = 'google/gemma-2b'
|
| 8 |
|
| 9 |
|
| 10 |
# Optional: use a cache directory
|
|
|
|
| 3 |
from transformers import pipeline, AutoTokenizer, AutoModelForCausalLM
|
| 4 |
import os
|
| 5 |
|
| 6 |
+
# MODEL_ID = "google/gemma-3n-E4B-it" # out of memory
|
| 7 |
+
MODEL_ID = 'google/gemma-2-2b-it'
|
| 8 |
|
| 9 |
|
| 10 |
# Optional: use a cache directory
|