LLM4Binary commited on
Commit
8b01405
·
verified ·
1 Parent(s): da3359f

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
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-6.7b-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
 
 
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