Update README.md
Browse files
README.md
CHANGED
@@ -72,7 +72,7 @@ for opt_state in OPT:
|
|
72 |
from transformers import AutoTokenizer, AutoModelForCausalLM
|
73 |
import torch
|
74 |
|
75 |
-
model_path = 'LLM4Binary/llm4decompile-
|
76 |
tokenizer = AutoTokenizer.from_pretrained(model_path)
|
77 |
model = AutoModelForCausalLM.from_pretrained(model_path,torch_dtype=torch.bfloat16).cuda()
|
78 |
|
|
|
72 |
from transformers import AutoTokenizer, AutoModelForCausalLM
|
73 |
import torch
|
74 |
|
75 |
+
model_path = 'LLM4Binary/llm4decompile-1.3b-v1.5' # V1.5 Model
|
76 |
tokenizer = AutoTokenizer.from_pretrained(model_path)
|
77 |
model = AutoModelForCausalLM.from_pretrained(model_path,torch_dtype=torch.bfloat16).cuda()
|
78 |
|