hicai-zju
/

InstructBioMol-base

Model card Files Files and versions

XiangZH commited on May 1

Commit

7fc5eb6

·

verified ·

1 Parent(s): 288a437

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -57,7 +57,7 @@ import torch
 model_name = "hicai-zju/InstructBioMol-base"
 tokenizer = LlamaTokenizer.from_pretrained(model_name)
-model = LlamaForCausalLM.from_pretrained(model_name, device_map="auto")
 prompt = "<p>M"  # protein sequence
 # prompt = "[C]"  # molecule sequence

 model_name = "hicai-zju/InstructBioMol-base"
 tokenizer = LlamaTokenizer.from_pretrained(model_name)
+model = LlamaForCausalLM.from_pretrained(model_name, device_map="cuda:0")
 prompt = "<p>M"  # protein sequence
 # prompt = "[C]"  # molecule sequence