arise-sustech
/

llm4decompile-33b

Text Generation

text-generation-inference

Inference Endpoints

Model card Files Files and versions Community

arise-sustech commited on Mar 1

Commit

27a5489

•

1 Parent(s): 86629f7

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -75,7 +75,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=200)
 c_func_decompile = tokenizer.decode(outputs[0][len(inputs[0]):-1])
 ```

     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=512)
 c_func_decompile = tokenizer.decode(outputs[0][len(inputs[0]):-1])
 ```