DataPilot commited on
Commit
62558db
·
verified ·
1 Parent(s): 59ba60d

instruction

Browse files
Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -24,8 +24,8 @@ import torch
24
  from transformers import AutoModelForSequenceClassification
25
  from transformers import AutoModelForCausalLM, AutoTokenizer, pipeline #transformerとtorchがインストールされていることを前提とします。
26
 
27
- model = AutoModelForCausalLM.from_pretrainedmodel = AutoModelForCausalLM.from_pretrained("DataPilot/ArrowSmart_1.7b_instraction")
28
- tokenizer = AutoTokenizer.from_pretrained("DataPilot/ArrowSmart_1.7b_instraction")
29
 
30
  generator = pipeline("text-generation", model=model, tokenizer=tokenizer, device=0)
31
 
 
24
  from transformers import AutoModelForSequenceClassification
25
  from transformers import AutoModelForCausalLM, AutoTokenizer, pipeline #transformerとtorchがインストールされていることを前提とします。
26
 
27
+ model = AutoModelForCausalLM.from_pretrainedmodel = AutoModelForCausalLM.from_pretrained("DataPilot/ArrowSmart_1.7b_instruction")
28
+ tokenizer = AutoTokenizer.from_pretrained("DataPilot/ArrowSmart_1.7b_instruction")
29
 
30
  generator = pipeline("text-generation", model=model, tokenizer=tokenizer, device=0)
31