MadeAgents
/

Hammer-1.5b

Model card Files Files and versions Community

linqq9 commited on 27 days ago

Commit

4fa6b2b

•

1 Parent(s): fa17972

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -41,7 +41,7 @@ import json
 import torch
 from transformers import AutoModelForCausalLM, AutoTokenizer
-model_name = "MadeAgents/Hammer-7b"
 model = AutoModelForCausalLM.from_pretrained(model_name, device_map="auto", torch_dtype="auto", trust_remote_code=True)
 tokenizer = AutoTokenizer.from_pretrained(model_name)

 import torch
 from transformers import AutoModelForCausalLM, AutoTokenizer
+model_name = "MadeAgents/Hammer-1.5b"
 model = AutoModelForCausalLM.from_pretrained(model_name, device_map="auto", torch_dtype="auto", trust_remote_code=True)
 tokenizer = AutoTokenizer.from_pretrained(model_name)