Osnly commited on
Commit
5c34dd5
·
verified ·
1 Parent(s): 66df884

Update src/plan.py

Browse files
Files changed (1) hide show
  1. src/plan.py +1 -1
src/plan.py CHANGED
@@ -6,7 +6,7 @@ import os
6
 
7
 
8
  hf_token = os.environ.get("HUGGINGFACE_TOKEN")
9
- model_id = "google/gemma-3n-E4B-it"
10
 
11
  cache_dir = "/tmp/hf_cache"
12
  tokenizer = AutoTokenizer.from_pretrained(model_id, token=hf_token, cache_dir=cache_dir)
 
6
 
7
 
8
  hf_token = os.environ.get("HUGGINGFACE_TOKEN")
9
+ model_id = "google/gemma-3n-E4B"
10
 
11
  cache_dir = "/tmp/hf_cache"
12
  tokenizer = AutoTokenizer.from_pretrained(model_id, token=hf_token, cache_dir=cache_dir)