AtakanTekparmak
commited on
fix: Fixed model name in usage
Browse files
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/
|
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 |
)
|