arise-sustech commited on
Commit
cbedf82
·
verified ·
1 Parent(s): 8791ca7

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
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-1.3b'
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