kaiku03 commited on
Commit
ec1520d
·
verified ·
1 Parent(s): 8ce1e75

Update tools/news_reporter.py

Browse files
Files changed (1) hide show
  1. 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" # confirm this exact repo id exists
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