AtakanTekparmak commited on
Commit
eff7ed5
·
verified ·
1 Parent(s): f1db0ea

fix: Fixed model name in usage

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -35,7 +35,7 @@ import json
35
  from typing import Any, Dict, List
36
  from transformers import AutoModelForCausalLM, AutoTokenizer
37
 
38
- model_name = "driaforall/function-calling-3B-30k-2e"
39
  model = AutoModelForCausalLM.from_pretrained(
40
  model_name, device_map="auto", torch_dtype="auto", trust_remote_code=True
41
  )
 
35
  from typing import Any, Dict, List
36
  from transformers import AutoModelForCausalLM, AutoTokenizer
37
 
38
+ model_name = "driaforall/Dria-Agent-a-3B"
39
  model = AutoModelForCausalLM.from_pretrained(
40
  model_name, device_map="auto", torch_dtype="auto", trust_remote_code=True
41
  )