pfnet
/

plamo-13b-instruct

Text Generation

Model card Files Files and versions

torotoki commited on Oct 30, 2023

Commit

c167a74

·

1 Parent(s): 7fbb404

Update README.md

Files changed (1) hide show

README.md +6 -6

README.md CHANGED Viewed

@@ -26,14 +26,14 @@ Install the required libraries as follows:
 Execute the following python code:
 ```python
 tokenizer = AutoTokenizer.from_pretrained(
-  "pfnet/plamo-13b-instruct",
-  trust_remote_code=True,
 )
 model = AutoModelForCausalLM.from_pretrained(
-  "pfnet/plamo-13b-instruct",
-  trust_remote_code=True,
-  torch_dtype=torch.bfloat16,
-  device_map="auto",
 )
 def completion(prompt: str, max_new_tokens: int = 128) -> str:

 Execute the following python code:
 ```python
 tokenizer = AutoTokenizer.from_pretrained(
+    "pfnet/plamo-13b-instruct",
+    trust_remote_code=True,
 )
 model = AutoModelForCausalLM.from_pretrained(
+    "pfnet/plamo-13b-instruct",
+    trust_remote_code=True,
+    torch_dtype=torch.bfloat16,
+    device_map="auto",
 )
 def completion(prompt: str, max_new_tokens: int = 128) -> str: