Update README.md
Browse files
README.md
CHANGED
|
@@ -67,7 +67,7 @@ Then use LLM4Decompile to translate the assembly instructions into C:
|
|
| 67 |
from transformers import AutoTokenizer, AutoModelForCausalLM
|
| 68 |
import torch
|
| 69 |
|
| 70 |
-
model_path = 'arise-sustech/llm4decompile-33b'
|
| 71 |
tokenizer = AutoTokenizer.from_pretrained(model_path)
|
| 72 |
model = AutoModelForCausalLM.from_pretrained(model_path,torch_dtype=torch.bfloat16).cuda()
|
| 73 |
|
|
|
|
| 67 |
from transformers import AutoTokenizer, AutoModelForCausalLM
|
| 68 |
import torch
|
| 69 |
|
| 70 |
+
model_path = 'arise-sustech/llm4decompile-33b' #note: it may need multi-gpu support
|
| 71 |
tokenizer = AutoTokenizer.from_pretrained(model_path)
|
| 72 |
model = AutoModelForCausalLM.from_pretrained(model_path,torch_dtype=torch.bfloat16).cuda()
|
| 73 |
|