ejschwartz commited on
Commit
073c7e0
·
1 Parent(s): d2a677d

add field decoder

Browse files
Files changed (1) hide show
  1. app.py +3 -0
app.py CHANGED
@@ -41,6 +41,9 @@ tokenizer = AutoTokenizer.from_pretrained(
41
  vardecoder_model = AutoModelForCausalLM.from_pretrained(
42
  "ejschwartz/resym-vardecoder", torch_dtype=torch.bfloat16, device_map="auto"
43
  )
 
 
 
44
 
45
  example = r"""{
46
  "input": "What are the original name and data type of variables `a1`, `a2`, `a3`, `v4`, `v5`?\n```\n__int64 __fastcall sub_410D81(__int64 a1, __int64 a2, __int64 a3)\n{\nint v4; // [rsp+20h] [rbp-20h] BYREF\n__int64 v5; // [rsp+28h] [rbp-18h]\n\nif ( !a1 || !a2 || !a3 )\nreturn 0LL;\nv4 = 5;\nv5 = a3;\nreturn sub_411142(a1, a2, &v4);\n}\n```",
 
41
  vardecoder_model = AutoModelForCausalLM.from_pretrained(
42
  "ejschwartz/resym-vardecoder", torch_dtype=torch.bfloat16, device_map="auto"
43
  )
44
+ fielddecoder_model = AutoModelForCausalLM.from_pretrained(
45
+ "ejschwartz/resym-vardecoder", torch_dtype=torch.bfloat16, device_map="auto"
46
+ )
47
 
48
  example = r"""{
49
  "input": "What are the original name and data type of variables `a1`, `a2`, `a3`, `v4`, `v5`?\n```\n__int64 __fastcall sub_410D81(__int64 a1, __int64 a2, __int64 a3)\n{\nint v4; // [rsp+20h] [rbp-20h] BYREF\n__int64 v5; // [rsp+28h] [rbp-18h]\n\nif ( !a1 || !a2 || !a3 )\nreturn 0LL;\nv4 = 5;\nv5 = a3;\nreturn sub_411142(a1, a2, &v4);\n}\n```",