arise-sustech
commited on
Update README.md
Browse files
README.md
CHANGED
@@ -69,7 +69,7 @@ Then use LLM4Decompile to translate the assembly instructions into C:
|
|
69 |
from transformers import AutoTokenizer, AutoModelForCausalLM
|
70 |
import torch
|
71 |
|
72 |
-
model_path = 'arise-sustech/llm4decompile-
|
73 |
tokenizer = AutoTokenizer.from_pretrained(model_path)
|
74 |
model = AutoModelForCausalLM.from_pretrained(model_path,torch_dtype=torch.bfloat16).cuda()
|
75 |
|
|
|
69 |
from transformers import AutoTokenizer, AutoModelForCausalLM
|
70 |
import torch
|
71 |
|
72 |
+
model_path = 'arise-sustech/llm4decompile-6.7b-nsp'
|
73 |
tokenizer = AutoTokenizer.from_pretrained(model_path)
|
74 |
model = AutoModelForCausalLM.from_pretrained(model_path,torch_dtype=torch.bfloat16).cuda()
|
75 |
|