Hugging Face's logo

Spaces:
masanorihirano
/
plamo-2-1b
Running

App Files Community
masanorihirano commited on
Commit
65c07aa
·
verified ·
1 Parent(s): 6cdcba1

Update model_pull.py

Browse files
Files changed (1) hide show
  1. model_pull.py +2 -6
model_pull.py CHANGED
@@ -1,7 +1,3 @@
1
- from transformers import AutoModelForCausalLM
2
- from transformers import AutoTokenizer
3
 
4
- BASE_MODEL = "pfnet/plamo-2-1b"
5
-
6
- tokenizer = AutoTokenizer.from_pretrained(BASE_MODEL, trust_remote_code=True)
7
- # model = AutoModelForCausalLM.from_pretrained(BASE_MODEL, trust_remote_code=True)
 
1
+ from huggingface_hub import snapshot_download
 
2
 
3
+ snapshot_download(repo_id="pfnet/plamo-2-1b")