losyer8 commited on
Commit
1189142
1 Parent(s): 5541839

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -55,8 +55,8 @@ Checkpoints format: `transformers` (Megatron-DeepSpeed format available [here](h
55
  ```python
56
  import torch
57
  from transformers import AutoTokenizer, AutoModelForCausalLM
58
- tokenizer = AutoTokenizer.from_pretrained("llm-jp/[Model_Name]")
59
- model = AutoModelForCausalLM.from_pretrained("llm-jp/[Model_Name]", torch_dtype=torch.float16)
60
  text = "自然言語処理とは何か"
61
  text = text + "### 回答:"
62
  tokenized_input = tokenizer.encode(text, add_special_tokens=False, return_tensors="pt").to(model.device)
 
55
  ```python
56
  import torch
57
  from transformers import AutoTokenizer, AutoModelForCausalLM
58
+ tokenizer = AutoTokenizer.from_pretrained("llm-jp/llm-jp-13b-instruct-full-jaster-dolly-oasst-v1.0")
59
+ model = AutoModelForCausalLM.from_pretrained("llm-jp/llm-jp-13b-instruct-full-jaster-dolly-oasst-v1.0", torch_dtype=torch.float16)
60
  text = "自然言語処理とは何か"
61
  text = text + "### 回答:"
62
  tokenized_input = tokenizer.encode(text, add_special_tokens=False, return_tensors="pt").to(model.device)