Hugging Face's logo

pfnet
/
plamo-13b-instruct

Text Generation
Transformers
Safetensors
English
Japanese
plamo
conversational
custom_code
Model card Files Community
1
torotoki commited on
Commit
c167a74
•
1 Parent(s): 7fbb404

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +6 -6
README.md CHANGED
@@ -26,14 +26,14 @@ Install the required libraries as follows:
26
  Execute the following python code:
27
  ```python
28
  tokenizer = AutoTokenizer.from_pretrained(
29
- "pfnet/plamo-13b-instruct",
30
- trust_remote_code=True,
31
  )
32
  model = AutoModelForCausalLM.from_pretrained(
33
- "pfnet/plamo-13b-instruct",
34
- trust_remote_code=True,
35
- torch_dtype=torch.bfloat16,
36
- device_map="auto",
37
  )
38
 
39
  def completion(prompt: str, max_new_tokens: int = 128) -> str:
 
26
  Execute the following python code:
27
  ```python
28
  tokenizer = AutoTokenizer.from_pretrained(
29
+ "pfnet/plamo-13b-instruct",
30
+ trust_remote_code=True,
31
  )
32
  model = AutoModelForCausalLM.from_pretrained(
33
+ "pfnet/plamo-13b-instruct",
34
+ trust_remote_code=True,
35
+ torch_dtype=torch.bfloat16,
36
+ device_map="auto",
37
  )
38
 
39
  def completion(prompt: str, max_new_tokens: int = 128) -> str: