Neo111x
/

falcon3-decompiler-7b-v1

Model card Files Files and versions Community

Neo111x commited on 10 days ago

Commit

5945d88

•

1 Parent(s): 4956fc5

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -23,7 +23,7 @@ import os
 model_path = 'Neo111x/falcon3-decompiler-7b-v1' # V1.5 Model
 tokenizer = AutoTokenizer.from_pretrained(model_path)
-model = AutoModelForCausalLM.from_pretrained(model_path,torch_dtype=torch.bfloat16).cuda()
 asm_func = """

 model_path = 'Neo111x/falcon3-decompiler-7b-v1' # V1.5 Model
 tokenizer = AutoTokenizer.from_pretrained(model_path)
+model = AutoModelForCausalLM.from_pretrained(model_path,torch_dtype=torch.bfloat16)
 asm_func = """