Update README.md

README.md

@@ -72,7 +72,7 @@ Then use LLM4Decompile to translate the assembly instructions into C:
 from transformers import AutoTokenizer, AutoModelForCausalLM
 import torch
 
-model_path = 'arise-sustech/llm4decompile-1.3b'
+model_path = 'arise-sustech/llm4decompile-6.7b-uo'
 tokenizer = AutoTokenizer.from_pretrained(model_path)
 model = AutoModelForCausalLM.from_pretrained(model_path,torch_dtype=torch.bfloat16).cuda()
 
@@ -80,7 +80,7 @@ with open(fileName +'_' + opt_state +'.asm','r') as f:#original file
     asm_func = f.read()
 inputs = tokenizer(asm_func, return_tensors="pt").to(model.device)
 with torch.no_grad():
-    outputs = model.generate(**inputs, max_new_tokens=500)
+    outputs = model.generate(**inputs, max_new_tokens=512)
 c_func_decompile = tokenizer.decode(outputs[0][len(inputs[0]):-1])
 ```