Spaces:
Runtime error
Runtime error
Commit
·
073c7e0
1
Parent(s):
d2a677d
add field decoder
Browse files
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```",
|